diff --git a/laborious/utils/repository/model_repository.py b/laborious/utils/repository/model_repository.py index 20ac9f3..1ab72c1 100644 --- a/laborious/utils/repository/model_repository.py +++ b/laborious/utils/repository/model_repository.py @@ -768,6 +768,8 @@ class MLFlowRepository: data_path = f'{model_temp_path}/retrain_data.csv' + makedirs(model_temp_path, exist_ok=True) + data.to_csv(data_path, index=True) self.logger.custom_info(