correct mkdir path

Godwin-T · Dec 20, 2023 · 205dc78 · 205dc78
1 parent 66b63f4
commit 205dc78
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/scripts/preprocess_data.py b/scripts/preprocess_data.py
@@ -104,9 +104,9 @@ def main():
     # Impute & Write processed dataset
     weather_features_processed = impute_data(weather)
 
-    if not os.path.dirname(PROCESSED_DATASET):
-        os.mkdir(os.path.dirname())
-        
+    if not os.path.exists('./processed_data'):
+        os.mkdir('./processed_data')
+
     weather_features_processed.to_csv(PROCESSED_DATASET, index=None)
 
 if __name__ == "__main__":