Skip to content

Commit 5763b79

Browse files
committed
feat: model version retrieval testing
1 parent e23d6a0 commit 5763b79

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/webapp/databricks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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
)

src/webapp/routers/models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -592,8 +592,8 @@ def trigger_inference_run(
592592
"/{inst_id}/models/{model_name}/get-model-versions"
593593
)
594594
def 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)],

0 commit comments

Comments
 (0)