fix test

Signed-off-by: yangxuan <[email protected]>

tests/test_rate_runner.py

@@ -63,7 +63,7 @@ def get_db(db: str, config: dict) -> VectorDB:
     parser = argparse.ArgumentParser()
     parser.add_argument("-r", "--insert_rate", type=int, default="1000", help="insert entity row count per seconds, cps")
     parser.add_argument("-d", "--db", type=str, default=DB.Milvus.name, help="db name")
-    parser.add_argument("-t", "--duration", type=int, default=120, help="stage search duration in seconds")
+    parser.add_argument("-t", "--duration", type=int, default=300, help="stage search duration in seconds")
     parser.add_argument("--use_s3", action='store_true', help="whether to use S3 dataset")
 
     flags = parser.parse_args()
@@ -84,5 +84,5 @@ def get_db(db: str, config: dict) -> VectorDB:
         insert_rate=flags.insert_rate,
         conc=conc,
         search_stage=search_stage,
-        stage_search_dur=flags.duration,
+        read_dur_after_write=flags.duration,
         local=flags.use_s3)