[Doc] Fix the type error in example Notebook 2 (#886)

*Issue #, if available:* *Description of changes:* This PR fix the type error of `reverse_edge_types_map` in the API example notebook 2. When it should be a dictionary. When set the `exclude_training_targets` to be `True`, this map will be used. By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice. Co-authored-by: Ubuntu <[email protected]>
awslabs · Jun 20, 2024 · 07cc472 · 07cc472
1 parent 285bbaa
commit 07cc472
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/source/notebooks/Notebook_2_LP_Pipeline.ipynb b/docs/source/notebooks/Notebook_2_LP_Pipeline.ipynb
@@ -94,8 +94,8 @@
     "    num_negative_edges=10,\n",
     "    node_feats=nfeats_4_modeling,\n",
     "    batch_size=64,\n",
-    "    exclude_training_targets=False,\n",
-    "    reverse_edge_types_map=[\"paper,citing,cited,paper\"],\n",
+    "    exclude_training_targets=True,\n",
+    "    reverse_edge_types_map={(\"paper\", \"citing\", \"paper\"):(\"paper\",\"cited\",\"paper\")},\n",
     "    train_task=True)\n",
     "val_dataloader = gs.dataloading.GSgnnLinkPredictionTestDataLoader(\n",
     "    dataset=acm_data,\n",