added download equals True

rapidsai · Sep 28, 2023 · 5fcf904 · 5fcf904
1 parent 0972bd4
commit 5fcf904
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/cugraph/cugraph/tests/layout/test_force_atlas2.py b/python/cugraph/cugraph/tests/layout/test_force_atlas2.py
@@ -146,7 +146,7 @@ def on_train_end(self, positions):
 @pytest.mark.parametrize("max_iter", MAX_ITERATIONS)
 @pytest.mark.parametrize("barnes_hut_optimize", BARNES_HUT_OPTIMIZE)
 def test_force_atlas2(graph_file, score, max_iter, barnes_hut_optimize):
-    cu_M = graph_file.get_edgelist()
+    cu_M = graph_file.get_edgelist(download=True)
     test_callback = TestCallback()
     cu_pos = cugraph_call(
         cu_M,
@@ -177,10 +177,10 @@ def test_force_atlas2(graph_file, score, max_iter, barnes_hut_optimize):
     """
 
     if "string" in graph_file.metadata["col_types"]:
-        df = renumbered_edgelist(graph_file.get_edgelist())
+        df = renumbered_edgelist(graph_file.get_edgelist(download=True))
         M = get_coo_array(df)
     else:
-        M = get_coo_array(graph_file.get_edgelist())
+        M = get_coo_array(graph_file.get_edgelist(download=True))
     cu_trust = trustworthiness(M, cu_pos[["x", "y"]].to_pandas())
     print(cu_trust, score)
     assert cu_trust > score