SIENTIAPDE-1241: refactor train_model workflow due to I/O errors.
This commit is contained in:
12
model_manager/utils/exceptions.py
Normal file
12
model_manager/utils/exceptions.py
Normal file
@@ -0,0 +1,12 @@
|
||||
class ModelTrainingError(Exception):
|
||||
def __init__(self, model_trained: bool, model_saved: bool, message: str | None = None):
|
||||
self.model_trained = model_trained
|
||||
self.model_saved = model_saved
|
||||
|
||||
if message is None:
|
||||
message = (
|
||||
'Model training workflow failed '
|
||||
f'(model_trained={model_trained}, model_saved={model_saved})'
|
||||
)
|
||||
|
||||
super().__init__(message)
|
||||
Reference in New Issue
Block a user