From 36b5f921e9c0a4dd5da3a66f21e8f6ba7ac3f61d Mon Sep 17 00:00:00 2001 From: Xiang Song Date: Thu, 30 Nov 2023 23:03:44 -0800 Subject: [PATCH] Update --- tests/unit-tests/gconstruct/test_remap_result.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/tests/unit-tests/gconstruct/test_remap_result.py b/tests/unit-tests/gconstruct/test_remap_result.py index 7e908150dc..1421c8f5b5 100644 --- 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()