fix for empty stresses df

pastas · Oct 10, 2024 · a95cc1d · a95cc1d
1 parent 1cf6332
commit a95cc1d
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/pastastore/extensions/hpd.py b/pastastore/extensions/hpd.py
@@ -580,6 +580,17 @@ def update_knmi_meteo(
         **kwargs : dict, optional
             Additional keyword arguments to pass to `hpd.read_knmi()`
         """
+        if "source" not in self._store.stresses.columns:
+            msg = (
+                "Cannot update KNMI stresses! "
+                "KNMI stresses cannot be identified if 'source' column is not defined."
+            )
+            logger.error(msg)
+            if raise_on_error:
+                raise ValueError(msg)
+            else:
+                return
+
         if names is None:
             names = self._store.stresses.loc[
                 self._store.stresses["source"] == "KNMI"