feat: added model deletion endpoint

Author: Mesh-ach

src/webapp/databricks.py

@@ -533,7 +533,7 @@ def fetch_model_version(self, catalog_name: str, inst_name: str, model_name: str
         latest_version = max(model_versions, key=lambda v: int(v.version))
 
         return latest_version
-    
+
     def delete_model(self, catalog_name: str, inst_name: str, model_name: str):
         schema = databricksify_inst_name(inst_name)
         model_name_path = f"{catalog_name}.{schema}_gold.{model_name}"

src/webapp/routers/models.py

@@ -310,6 +310,7 @@ def read_inst_model(
         "framework": query_result[0][0].framework,
     }
 
+
 @router.delete("/{inst_id}/models/{model_name}", response_model=ModelInfo)
 def delete_model(
     inst_id: str,
@@ -319,26 +320,23 @@ def delete_model(
     sql_session: Annotated[Session, Depends(get_session)],
     databricks_control: Annotated[DatabricksControl, Depends(DatabricksControl)],
 ) -> Any:
-    
     transformed_model_name = str(decode_url_piece(model_name)).strip()
     has_access_to_inst_or_err(inst_id, current_user)
     model_owner_and_higher_or_err(current_user, "modify batch")
 
     local_session.set(sql_session)
     sess = local_session.get()
 
-    query_result = (
-        sess.execute(
-            select(InstTable).where(
-                InstTable.id == str_to_uuid(inst_id)))
-        .all()
-    )
+    query_result = sess.execute(
+        select(InstTable).where(InstTable.id == str_to_uuid(inst_id))
+    ).all()
 
     model_list = sess.execute(
-        select(ModelTable.where(
-            ModelTable.name == str_to_uuid(model_name),
-            ModelTable.inst_id == str_to_uuid(inst_id),
-        )
+        select(
+            ModelTable.where(
+                ModelTable.name == str_to_uuid(model_name),
+                ModelTable.inst_id == str_to_uuid(inst_id),
+            )
         )
     ).scalar_one_or_none()
     if model_list is None:
@@ -362,6 +360,7 @@ def delete_model(
         "deleted_from_databricks": delete_from_databricks,
     }
 
+
 @router.get("/{inst_id}/models/{model_name}/runs", response_model=list[RunInfo])
 def read_inst_model_outputs(
     inst_id: str,