From 0b79b2bdd1aa4421a0d8070440b54fbe85df283c Mon Sep 17 00:00:00 2001 From: Shreya Shankar Date: Fri, 23 Aug 2024 00:49:11 -0700 Subject: [PATCH] Fix tests --- tests/test_basic.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/test_basic.py b/tests/test_basic.py index 3a0b6372..8b07f62d 100644 --- 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)