Temporarily skip deleting copied dask dataframe to avoid crash

rapidsai · Sep 22, 2023 · 0d8ff97 · 0d8ff97
1 parent bd85881
commit 0d8ff97
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/python/cugraph/cugraph/tests/traversal/test_bfs_mg.py b/python/cugraph/cugraph/tests/traversal/test_bfs_mg.py
@@ -21,8 +21,6 @@
 import cugraph.dask as dcg
 from cugraph.testing.utils import RAPIDS_DATASET_ROOT_DIR_PATH
 
-import random
-
 # =============================================================================
 # Pytest Setup / Teardown - called for each test function
 # =============================================================================
@@ -63,9 +61,7 @@ def modify_dataset(df):
         return cudf.concat([df, temp_df])
 
     meta = ddf._meta
-    ddf = ddf.map_partitions(
-        modify_dataset, meta=meta, token="custom-" + str(random.random())
-    )
+    ddf = ddf.map_partitions(modify_dataset, meta=meta)
 
     df = cudf.read_csv(
         input_data_path,