fix: change path with correct file name

octo-technology · Dec 12, 2023 · 5235296 · 5235296
1 parent 3361c70
commit 5235296
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dags/config.py b/dags/config.py
@@ -6,9 +6,9 @@
 GENERATED_DATA_FOLDER = os.path.join(DATA_FOLDER, "generated_data")
 PREDICTIONS_FOLDER = os.path.join(DATA_FOLDER, 'predictions')
 
-TRAIN_DATA_PATH = os.path.join(DATA_FOLDER, 'la-haute-borne-data-2013-2016.csv')
+TRAIN_DATA_PATH = os.path.join(DATA_FOLDER, 'la-haute-borne-data-2017-2020.csv')
 # TODO change, should point to generator
-GENERATED_DATA_PATH = os.path.join(DATA_FOLDER, 'la-haute-borne-data-2013-2016.csv')
+GENERATED_DATA_PATH = os.path.join(DATA_FOLDER, 'la-haute-borne-data-2017-2020.csv')
 
 FEATURES_PATH = os.path.join(DATA_FOLDER, 'prepared_features.parquet')