- Updated the `Training` class to raise `ModelTrainingError` on training failures for better error management. - Enhanced the `run` method in `TrainModel` to return training results and ensure proper resource cleanup, including validation files. - Refactored exception handling to prevent silent failures during resource cleanup and experiment run updates. - Adjusted type hints for improved clarity and consistency in method signatures.
15 KiB
15 KiB