[Doc] Fix WeightBasis documentation (dmlc#3189)

* fix * fix type Co-authored-by: Minjie Wang <[email protected]>
ammuaj · Jul 28, 2021 · 2583ec5 · 2583ec5
1 parent cb4a83a
commit 2583ec5
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/docs/source/api/python/nn.pytorch.rst b/docs/source/api/python/nn.pytorch.rst
@@ -275,6 +275,13 @@ Sequential
     :members:
     :show-inheritance:
 
+WeightBasis
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. autoclass:: dgl.nn.pytorch.utils.WeightBasis
+    :members:
+    :show-inheritance:
+
 KNNGraph
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 

diff --git a/examples/pytorch/seal/main.py b/examples/pytorch/seal/main.py
@@ -85,7 +85,7 @@ def main(args, print_fn=print):
                          subsample_ratio=args.subsample_ratio, use_coalesce=use_coalesce, prefix=args.dataset,
                          save_dir=args.save_dir, num_workers=args.num_workers, print_fn=print_fn)
     node_attribute = seal_data.ndata['feat']
-    edge_weight = seal_data.edata['edge_weight'].float()
+    edge_weight = seal_data.edata['weight'].float()
 
     train_data = seal_data('train')
     val_data = seal_data('valid')