Fix tests

ucbepic · Aug 23, 2024 · 0b79b2b · 0b79b2b
1 parent d97ee36
commit 0b79b2b
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/test_basic.py b/tests/test_basic.py
@@ -398,9 +398,7 @@ def test_gather_operation(
 def test_split_gather_combined(
     split_config, gather_config, default_model, max_threads, sample_data
 ):
-    split_op = SplitOperation(
-        split_config, default_model, max_threads, name="split_doc"
-    )
+    split_op = SplitOperation(split_config, default_model, max_threads)
     gather_op = GatherOperation(gather_config, default_model, max_threads)
 
     split_results, split_cost = split_op.execute(sample_data)