fix: formatting style

Author: Mesh-ach

src/webapp/databricks.py

@@ -523,11 +523,16 @@ def fetch_model_version(self, catalog_name: str, inst_name: str, model_name: str
             )
             raise ValueError(f"setup_new_inst(): Workspace client creation failed: {e}")
 
-        model_info = w.model_versions.list(
+        model_versions = list(w.model_versions.list(
             full_name=model_name_path,
-        )
+        ))
+
+        if not model_versions:
+            raise ValueError(f"No versions found for model: {model_name_path}")
+
+        latest_version = max(model_versions, key=lambda v: int(v.version))
 
-        return model_info
+        return latest_version
 
     def get_key_for_file(
         self, mapping: Dict[str, Any], file_name: str

src/webapp/routers/models.py

@@ -622,10 +622,10 @@ def get_model_versions(
     print(f"Initial model name = {model_name}")
     print(f"Converted model name {transformed_model_name}")
 
-    model_version_info = databricks_control.fetch_model_version(
+    latest_model_version = databricks_control.fetch_model_version(
         catalog_name=env_vars["CATALOG_NAME"],
         inst_name=f"{query_result[0][0].name}",
         model_name=transformed_model_name,
     )
 
-    return model_version_info
+    return latest_model_version