fix. (#2907)

Co-authored-by: Zheng <dzzhen@3c22fba32af5.ant.amazon.com>

fix. (#2907)
Co-authored-by: Zheng <dzzhen@3c22fba32af5.ant.amazon.com>
a90296aa · Da Zheng · GitHub · e156331b · a90296aa
Unverified Commit a90296aa authored May 13, 2021 by Da Zheng Committed by GitHub May 13, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

docs/source/guide/distributed-preprocessing.rst docs/source/guide/distributed-preprocessing.rst +1 -1

No files found.
--- a/docs/source/guide/distributed-preprocessing.rst
+++ b/docs/source/guide/distributed-preprocessing.rst
@@ -100,7 +100,7 @@ the mapping for each node type; the second dictionary contains the mapping for e
                                                         return_mapping=True)
    # Let's assume that node_emb is saved from the distributed training.
    orig_node_emb = th.zeros(node_emb.shape, dtype=node_emb.dtype)
-    orig_labels[node_map] = node_emb
+    orig_node_emb[node_map] = node_emb


 7.1.1 Distributed partitioning