fix: preserve dataframe dtypes

Fixes #38
resgroup · Nov 4, 2024 · b18e092 · b18e092
1 parent 980f6db
commit b18e092
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/wind_up/scada_funcs.py b/wind_up/scada_funcs.py
@@ -28,7 +28,8 @@
 
 
 def filter_stuck_data(df: pd.DataFrame) -> pd.DataFrame:
-    diffdf = df.groupby("TurbineName", observed=False).ffill().fillna(0).diff()
+    original_data_types = df.dtypes.to_dict()
+    diffdf = df.groupby("TurbineName", observed=False).ffill().fillna(0).astype(original_data_types).diff()
     stuck_data = (diffdf == 0).all(axis=1)
     very_low_wind_threshold = 1.5
     very_low_wind = df["WindSpeedMean"] < very_low_wind_threshold