Skip to content

Commit 72af8b9

Browse files
committed
Remove redundant casts, seems like my local mypy was acting up
1 parent cfca24e commit 72af8b9

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

vec_inf/client/_helper.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ def _get_model_configuration(self) -> ModelConfig:
104104

105105
def _get_launch_params(self) -> dict[str, Any]:
106106
"""Merge config defaults with CLI overrides."""
107-
params = cast(dict[str, Any], self.model_config.model_dump())
107+
params = self.model_config.model_dump()
108108

109109
# Process boolean fields
110110
for bool_field in BOOLEAN_FIELDS:
@@ -242,7 +242,7 @@ def _get_raw_status_output(self) -> str:
242242
output, stderr = utils.run_bash_command(status_cmd)
243243
if stderr:
244244
raise SlurmJobError(f"Error: {stderr}")
245-
return cast(str, output)
245+
return output
246246

247247
def _get_base_status_data(self) -> dict[str, Union[str, None]]:
248248
"""Extract basic job status information from scontrol output."""

vec_inf/client/api.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ def list_models(self) -> list[ModelInfo]:
7373
"""
7474
try:
7575
model_registry = ModelRegistry()
76-
return cast(list[ModelInfo], model_registry.get_all_models())
76+
return model_registry.get_all_models()
7777
except Exception as e:
7878
raise APIError(f"Failed to list models: {str(e)}") from e
7979

0 commit comments

Comments
 (0)