Update

awslabs · Dec 1, 2023 · 36b5f92 · 36b5f92
1 parent cc529c8
commit 36b5f92
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/unit-tests/gconstruct/test_remap_result.py b/tests/unit-tests/gconstruct/test_remap_result.py
@@ -170,12 +170,10 @@ def test_worker_remap_edge_pred():
 
         worker_remap_edge_pred(pred_path, src_nid_path, dst_nid_path,
                                ntypes[0], ntypes[1], output_path_prefix,
-                               chunk_size, write_data_parquet_file,
-                               preserve_input=True)
+                               chunk_size, write_data_parquet_file)
         worker_remap_edge_pred(pred_path, src_nid_path, dst_nid_path,
                                ntypes[0], ntypes[1], output_path_prefix,
-                               chunk_size, partial(write_data_csv_file, delimiter=","),
-                               preserve_input=True)
+                               chunk_size, partial(write_data_csv_file, delimiter=","))
         def read_csv(file, delimiter=","):
             data = pd.read_csv(file, delimiter=delimiter)
             src_nid = data["src_nid"].to_numpy()