[KVStore] Fix a small bug (#1686)

* update * update * update * update * update

[KVStore] Fix a small bug (#1686)
* update * update * update * update * update
7afd1d0d · Chao Ma · GitHub · 86c243b4 · 7afd1d0d
Unverified Commit 7afd1d0d authored Jun 23, 2020 by Chao Ma Committed by GitHub Jun 23, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

python/dgl/distributed/kvstore.py python/dgl/distributed/kvstore.py +1 -1

No files found.
--- a/python/dgl/distributed/kvstore.py
+++ b/python/dgl/distributed/kvstore.py
@@ -829,7 +829,7 @@ class KVClient(object):
            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