fix: formatting style

Author: Mesh-ach

src/webapp/database.py

@@ -551,8 +551,8 @@ class JobTable(Base):
         String(VAR_CHAR_STANDARD_LENGTH), nullable=True
     )
     completed: Mapped[bool] = mapped_column(nullable=True)
-    framework: Mapped[str | None] = mapped_column(
-        String(VAR_CHAR_STANDARD_LENGTH), nullable=False, default="sklearn"
+    model_run_id: Mapped[str | None] = mapped_column(
+        String(VAR_CHAR_STANDARD_LENGTH), nullable=True
     )
 
 

src/webapp/databricks.py

@@ -523,9 +523,11 @@ 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_versions = list(w.model_versions.list(
-            full_name=model_name_path,
-        ))
+        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}")

src/webapp/routers/models.py

@@ -556,7 +556,6 @@ def trigger_inference_run(
         gcp_external_bucket_name=get_external_bucket_name(inst_id),
         # The institution email to which pipeline success/failure notifications will get sent.
         email=cast(str, current_user.email),
-        model_type=query_result[0][0].framework,
     )
     try:
         res = databricks_control.run_pdp_inference(db_req)
@@ -568,14 +567,19 @@ def trigger_inference_run(
             detail=f"Databricks run_pdp_inference error. Error = {str(e)}",
         ) from e
     triggered_timestamp = datetime.now()
+    latest_model_version = databricks_control.fetch_model_version(
+        catalog_name=env_vars["CATALOG_NAME"],
+        inst_name=inst_result[0][0].name,
+        model_name=model_name,
+    )
     job = JobTable(
         id=res.job_run_id,
         triggered_at=triggered_timestamp,
         created_by=str_to_uuid(current_user.user_id),
         batch_name=req.batch_name,
         model_id=query_result[0][0].id,
         output_valid=False,
-        framework=query_result[0][0].framework,
+        model_run_id=latest_model_version.run_id,
     )
     local_session.get().add(job)
     return {
@@ -586,7 +590,7 @@ def trigger_inference_run(
         "triggered_at": triggered_timestamp,
         "batch_name": req.batch_name,
         "output_valid": False,
-        "framework": query_result[0][0].framework,
+        "model_run_id": latest_model_version.run_id,
     }