[KVStore] Fix a small bug (dmlc#1686)

* update * update * update * update * update
lili0710432 · Jun 23, 2020 · 7afd1d0 · 7afd1d0
1 parent 86c243b
commit 7afd1d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/dgl/distributed/kvstore.py b/python/dgl/distributed/kvstore.py
@@ -829,7 +829,7 @@ def pull_handler(data_store, name, local_offset)
             for server_id in range(self._server_count):
                 rpc.send_request(server_id, request)
             # recv response from all the server nodes
-            for _ in range(self._server_namebook):
+            for _ in range(self._server_count):
                 response = rpc.recv_response()
                 assert response.msg == REGISTER_PULL_MSG
         self._pull_handlers[name] = func