Replaces DATASETS_UNDIRECTED list with list of specific dataset objs.

rapidsai · Nov 19, 2023 · 82d69e8 · 82d69e8
1 parent 7067cb0
commit 82d69e8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/cugraph/cugraph/tests/centrality/test_edge_betweenness_centrality_mg.py b/python/cugraph/cugraph/tests/centrality/test_edge_betweenness_centrality_mg.py
@@ -16,7 +16,7 @@
 
 import dask_cudf
 from pylibcugraph.testing.utils import gen_fixture_params_product
-from cugraph.datasets import DATASETS_UNDIRECTED
+from cugraph.datasets import karate, dolphins
 
 import cugraph
 import cugraph.dask as dcg
@@ -41,7 +41,7 @@ def setup_function():
 
 
 # email_Eu_core is too expensive to test
-datasets = DATASETS_UNDIRECTED
+datasets = [karate, dolphins]
 
 
 # =============================================================================