fix: abort training pipeline on dataset conversion failure

This commit is contained in:
Roland Tannous 2026-03-04 06:42:48 +00:00
commit 645d7d357a

View file

@ -468,6 +468,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")