fix flake8

julienfoenet · Oct 7, 2024 · de60847 · de60847
1 parent 48f7c2f
commit de60847
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/formation_indus_ds_avancee/train_and_predict.py b/formation_indus_ds_avancee/train_and_predict.py
@@ -20,6 +20,7 @@ def train_model(features: pd.DataFrame, model_registry_folder: str) -> None:
     model.fit(X, y)
     joblib.dump(model, os.path.join(model_registry_folder, 'model.joblib'))
 
+
 def predict_with_io(features_path: str, model_path: str, predictions_folder: str) -> None:
     features = pd.read_parquet(features_path)
     features = predict(features, model_path)