From 2b704221f7add28d0796229c77ce3256a917093b Mon Sep 17 00:00:00 2001 From: Roland Tannous Date: Wed, 4 Mar 2026 06:42:48 +0000 Subject: [PATCH] fix: abort training pipeline on dataset conversion failure --- studio/backend/core/training/trainer.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/studio/backend/core/training/trainer.py b/studio/backend/core/training/trainer.py index 5ce273f43c..f5b2f18245 100644 --- a/studio/backend/core/training/trainer.py +++ b/studio/backend/core/training/trainer.py @@ -482,6 +482,14 @@ class UnslothTrainer: print("Stopped during dataset formatting\n") return None + # Abort if dataset formatting/conversion failed + if not dataset_info.get("success", True): + errors = dataset_info.get("errors", []) + error_msg = "; ".join(errors) if errors else "Dataset formatting failed" + logger.error(f"Dataset conversion failed: {error_msg}") + self._update_progress(error=error_msg) + return None + self._update_progress(status_message=f"Dataset formatted and ready for training") print(f"Dataset formatted successfully\n")