13 lines
441 B
Python
13 lines
441 B
Python
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)
|