Fix dropna argument for DWD Mosmix and DMO

earthobservations · May 19, 2024 · 9f34002 · 9f34002
1 parent d01b060
commit 9f34002
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/CHANGELOG.rst b/CHANGELOG.rst
@@ -4,6 +4,8 @@ Changelog
 Development
 ***********
 
+- Fix `dropna` argument for DWD Mosmix and DMO
+
 0.84.0 (15.05.2024)
 *******************
 

diff --git a/wetterdienst/provider/dwd/dmo/api.py b/wetterdienst/provider/dwd/dmo/api.py
@@ -1138,6 +1138,9 @@ def query(self) -> Iterator[ValuesResult]:
             if not df.is_empty() and self.sr.start_date and self.sr.start_date != DwdForecastDate.LATEST:
                 df = df.filter(pl.col(Columns.DATE.value).is_between(self.sr.start_date, self.sr.end_date))
 
+            if self.sr.dropna:
+                df = df.drop_nulls(subset="value")
+
             if not self.sr.tidy:
                 df = self._widen_df(df)
 

diff --git a/wetterdienst/provider/dwd/mosmix/api.py b/wetterdienst/provider/dwd/mosmix/api.py
@@ -1073,6 +1073,9 @@ def query(self) -> Iterator[ValuesResult]:
             if not df.is_empty() and self.sr.start_date and self.sr.start_date != DwdForecastDate.LATEST:
                 df = df.filter(pl.col(Columns.DATE.value).is_between(self.sr.start_date, self.sr.end_date))
 
+            if self.sr.dropna:
+                df = df.drop_nulls(subset="value")
+
             if not self.sr.tidy:
                 df = self._widen_df(df)