Merge pull request #5 from VibhuJawa/VibhuJawa-patch-4

Update workflow.py
rapidsai · May 26, 2023 · bd54bac · bd54bac
2 parents 1bd4d79 + 1cde195
commit bd54bac
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/python/cugraph-dgl/examples/multi_trainer_MG_example/workflow.py b/python/cugraph-dgl/examples/multi_trainer_MG_example/workflow.py
@@ -228,8 +228,7 @@ def run_workflow(rank, devices, scheduler_address):
     # Load dummy first
     # because new environments
     # require dataset download
-    _ = load_dgl_dataset()
-    del _
+    load_dgl_dataset()
     dask_worker_devices = [5, 6]
     cluster = setup_cluster(dask_worker_devices)