We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 809d1db commit ae5fe8fCopy full SHA for ae5fe8f
1 file changed
src/webapp/routers/models.py
@@ -339,8 +339,15 @@ def delete_model(
339
# 2) Optionally Delete models from databricks itself
340
# TODO: Add databricks deletion functionality
341
342
- sess.delete(model_list)
343
- sess.commit()
+ try:
+ 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
+
351
return {
352
"inst_id": inst_id,
353
"model_name": transformed_model_name,
0 commit comments