From a95cc1da1cc29b2b63ca76fb495c89e1a1b601fa Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Dav=C3=ADd=20Brakenhoff?= <d.brakenhoff@artesia-water.nl>
Date: Thu, 10 Oct 2024 14:07:10 +0200
Subject: [PATCH] fix for empty stresses df

---
 pastastore/extensions/hpd.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/pastastore/extensions/hpd.py b/pastastore/extensions/hpd.py
index 59962c7..4a9cbfd 100644
--- 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"