Skip to content

Commit ae5fe8f

Browse files
committed
fix: added status
1 parent 809d1db commit ae5fe8f

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

src/webapp/routers/models.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -339,8 +339,15 @@ def delete_model(
339339
# 2) Optionally Delete models from databricks itself
340340
# TODO: Add databricks deletion functionality
341341

342-
sess.delete(model_list)
343-
sess.commit()
342+
try:
343+
sess.delete(model_list)
344+
sess.commit()
345+
except Exception as e:
346+
sess.rollback()
347+
raise HTTPException(
348+
status_code=500, detail=f"DB batch delete failed after file cleanup: {e}"
349+
)
350+
344351
return {
345352
"inst_id": inst_id,
346353
"model_name": transformed_model_name,

0 commit comments

Comments
 (0)