Unload pre-existing internal DF

rapidsai · Mar 13, 2024 · 2c77751 · 2c77751
1 parent b9fd96d
commit 2c77751
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/cugraph/cugraph/tests/centrality/test_eigenvector_centrality_mg.py b/python/cugraph/cugraph/tests/centrality/test_eigenvector_centrality_mg.py
@@ -52,6 +52,7 @@ def setup_function():
 def test_dask_mg_eigenvector_centrality(dask_client, dataset, directed):
     input_data_path = dataset.get_path()
     print(f"dataset={input_data_path}")
+    dataset.unload()
     ddf = dataset.get_dask_edgelist()
     dg = cugraph.Graph(directed=True)
     dg.from_dask_cudf_edgelist(ddf, "src", "dst", store_transposed=True)
@@ -96,8 +97,8 @@ def test_dask_mg_eigenvector_centrality(dask_client, dataset, directed):
 def test_dask_mg_eigenvector_centrality_transposed_false(dask_client):
     dataset = DATASETS[0]
 
+    dataset.unload()
     ddf = dataset.get_dask_edgelist()
-
     dg = cugraph.Graph(directed=True)
     dg.from_dask_cudf_edgelist(ddf, "src", "dst", store_transposed=False)