Merge branch 'master' into scale_shuffle

AIStream-Peelout · Dec 28, 2021 · 69b15a0 · 69b15a0
2 parents c0a4539 + 40d1ace
commit 69b15a0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/flood_forecast/preprocessing/temporal_feats.py b/flood_forecast/preprocessing/temporal_feats.py
@@ -19,6 +19,8 @@ def create_feature(key: str, value: str, df: pd.DataFrame, dt_column: str):
     """
     if key == "day_of_week":
         df[key] = df[dt_column].map(lambda x: x.weekday())
+    elif key == "minute":
+        df[key] = df[dt_column].map(lambda x: x.minute)
     elif key == "hour":
         df[key] = df[dt_column].map(lambda x: x.hour)
     elif key == "day":