Skip to content

Commit 0e3fee8

Browse files
fix ruff lint issue
Signed-off-by: min.tian <min.tian.cn@gmail.com>
1 parent be2eb09 commit 0e3fee8

File tree

3 files changed

+10
-7
lines changed

3 files changed

+10
-7
lines changed

vectordb_bench/backend/runner/serial_runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -205,7 +205,7 @@ def run_endlessness(self) -> int:
205205
raise LoadTimeoutError(self.timeout)
206206

207207
def run(self) -> int:
208-
count, dur = self._insert_all_batches()
208+
count, _ = self._insert_all_batches()
209209
return count
210210

211211

vectordb_bench/cli/cli.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -183,9 +183,9 @@ def get_custom_case_config(parameters: dict) -> dict:
183183
"with_gt": parameters["custom_dataset_with_gt"],
184184
},
185185
}
186-
elif parameters["case_type"] == "NewIntFilterPerformanceCase" :
186+
elif parameters["case_type"] == "NewIntFilterPerformanceCase":
187187
custom_case_config = {
188-
"dataset_with_size_type" : parameters["dataset_with_size_type"],
188+
"dataset_with_size_type": parameters["dataset_with_size_type"],
189189
"filter_rate": parameters["filter_rate"],
190190
}
191191
return custom_case_config
@@ -425,10 +425,12 @@ class CommonTypedDict(TypedDict):
425425
str,
426426
click.option(
427427
"--dataset-with-size-type",
428-
help="Dataset with size type for NewIntFilterPerformanceCase, you can use Medium Cohere (768dim, 1M)|Large Cohere (768dim, 10M)|Medium Bioasq (1024dim, 1M)|Large Bioasq (1024dim, 10M)|Large OpenAI (1536dim, 5M)|Medium OpenAI (1536dim, 500K)",
428+
help="Dataset with size type for NewIntFilterPerformanceCase, you can use Medium Cohere (768dim, 1M)|"
429+
"Large Cohere (768dim, 10M)|Medium Bioasq (1024dim, 1M)|Large Bioasq (1024dim, 10M)|"
430+
"Large OpenAI (1536dim, 5M)|Medium OpenAI (1536dim, 500K)",
429431
default="Medium Cohere (768dim, 1M)",
430432
show_default=True,
431-
)
433+
),
432434
]
433435
filter_rate: Annotated[
434436
float,
@@ -437,9 +439,10 @@ class CommonTypedDict(TypedDict):
437439
help="Filter rate for NewIntFilterPerformanceCase",
438440
default=0.01,
439441
show_default=True,
440-
)
442+
),
441443
]
442444

445+
443446
class HNSWBaseTypedDict(TypedDict):
444447
m: Annotated[int | None, click.option("--m", type=int, help="hnsw m")]
445448
ef_construction: Annotated[

vectordb_bench/interface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -279,7 +279,7 @@ def kill_proc_tree(
279279
p.send_signal(sig)
280280
except psutil.NoSuchProcess:
281281
pass
282-
gone, alive = psutil.wait_procs(children, timeout=timeout, callback=on_terminate)
282+
_, alive = psutil.wait_procs(children, timeout=timeout, callback=on_terminate)
283283

284284
for p in alive:
285285
log.warning(f"force killing child process: {p}")

0 commit comments

Comments
 (0)