fix

Signed-off-by: yangxuan <[email protected]>
zilliztech · Nov 11, 2024 · 7569c20 · 7569c20
1 parent 0fc9f62
commit 7569c20
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_rate_runner.py b/tests/test_rate_runner.py
@@ -30,7 +30,7 @@ def test_rate_runner(db, insert_rate):
     _, t = runner.run_with_rate()
     log.info(f"insert run done, time={t}")
 
-def test_read_write_runner(db, insert_rate, conc: list, search_stage: Iterable[float], stage_search_dur: int, local: bool=False):
+def test_read_write_runner(db, insert_rate, conc: list, search_stage: Iterable[float], read_dur_after_write: int, local: bool=False):
     cohere = Dataset.COHERE.manager(1_000_000)
     if local is True:
         source = DatasetSource.AliyunOSS
@@ -44,7 +44,7 @@ def test_read_write_runner(db, insert_rate, conc: list, search_stage: Iterable[f
         dataset=cohere,
         insert_rate=insert_rate,
         search_stage=search_stage,
-        stage_search_dur=stage_search_dur,
+        read_dur_after_write=read_dur_after_write,
         concurrencies=conc
     )
     rw_runner.run_read_write()