Skip to content

Commit c0f7250

Browse files
committed
fix: formatting style
1 parent de8b250 commit c0f7250

2 files changed

Lines changed: 9 additions & 12 deletions

File tree

src/webapp/databricks.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -505,9 +505,7 @@ def _consume_chunk(chunk_obj: Any) -> int | None:
505505
pass
506506
return records
507507

508-
def fetch_model_version(
509-
self,
510-
model_name: str):
508+
def fetch_model_version(self, model_name: str):
511509
try:
512510
w = WorkspaceClient(
513511
host=databricks_vars["DATABRICKS_HOST_URL"],
@@ -523,10 +521,10 @@ def fetch_model_version(
523521

524522
model_info = w.model_versions.list(
525523
full_name=model_name,
526-
)
524+
)
527525

528526
return model_info
529-
527+
530528
def get_key_for_file(
531529
self, mapping: Dict[str, Any], file_name: str
532530
) -> Optional[str]:

src/webapp/routers/models.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -588,20 +588,19 @@ def trigger_inference_run(
588588
"framework": query_result[0][0].framework,
589589
}
590590

591-
@router.get(
592-
"/{inst_id}/models/{model_name}/get-model-versions"
593-
)
591+
592+
@router.get("/{inst_id}/models/{model_name}/get-model-versions")
594593
def get_model_versions(
595594
inst_id: str,
596595
model_name: str,
597596
current_user: Annotated[BaseUser, Depends(get_current_active_user)],
598597
sql_session: Annotated[Session, Depends(get_session)],
599598
databricks_control: Annotated[DatabricksControl, Depends(DatabricksControl)],
600599
) -> Any:
601-
600+
602601
model_name = decode_url_piece(model_name)
603602
has_access_to_inst_or_err(inst_id, current_user)
604-
603+
605604
model_version_info = databricks_control.fetch_model_version(model_name)
606-
607-
return model_version_info
605+
606+
return model_version_info

0 commit comments

Comments
 (0)