Skip to content

Commit b081625

Browse files
committed
Fix mypy errors
1 parent 182b0cc commit b081625

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vec_inf/cli/_helper.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -390,14 +390,14 @@ def _build_metrics_url(self) -> str:
390390
def _check_prefix_caching(self) -> bool:
391391
"""Check if prefix caching is enabled."""
392392
job_json = utils.read_slurm_log(
393-
self.status_info["model_name"],
393+
cast(str, self.status_info["model_name"]),
394394
self.slurm_job_id,
395395
"json",
396396
self.log_dir,
397397
)
398398
if isinstance(job_json, str):
399399
return False
400-
return job_json.get("enable_prefix_caching", False)
400+
return bool(cast(dict[str, str], job_json).get("enable_prefix_caching", False))
401401

402402
def fetch_metrics(self) -> Union[dict[str, float], str]:
403403
"""Fetch metrics from the endpoint."""

0 commit comments

Comments
 (0)