File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed
Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -521,7 +521,7 @@ def fetch_model_version(
521521 )
522522 raise ValueError (f"setup_new_inst(): Workspace client creation failed: { e } " )
523523
524- model_info = w .registered_models .list (
524+ model_info = w .model_versions .list (
525525 full_name = model_name ,
526526 include_aliases = True ,
527527 )
Original file line number Diff line number Diff line change @@ -592,8 +592,8 @@ def trigger_inference_run(
592592 "/{inst_id}/models/{model_name}/get-model-versions"
593593)
594594def get_model_versions (
595- model_name : str ,
596595 inst_id : str ,
596+ model_name : str ,
597597 current_user : Annotated [BaseUser , Depends (get_current_active_user )],
598598 sql_session : Annotated [Session , Depends (get_session )],
599599 databricks_control : Annotated [DatabricksControl , Depends (DatabricksControl )],
You can’t perform that action at this time.
0 commit comments