Use wind_speed_max for wind_gust and add wind direction avgs

sdss · Aug 24, 2024 · 22cab44 · 22cab44
1 parent 0512638
commit 22cab44
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/lvmapi/tools/weather.py b/src/lvmapi/tools/weather.py
@@ -118,7 +118,14 @@ async def get_weather_data(
             for ws in window_sizes
         },
         **{
-            f"wind_gust_{ws}": polars.col.wind_speed_avg.rolling_max_by(
+            f"wind_gust_{ws}": polars.col.wind_speed_max.rolling_max_by(
+                by="ts",
+                window_size=ws,
+            )
+            for ws in window_sizes
+        },
+        **{
+            f"wind_dir_avg_{ws}": polars.col.wind_dir_avg.rolling_mean_by(
                 by="ts",
                 window_size=ws,
             )