File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff 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."""
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments