Update src/frequenz/lib/notebooks/reporting/data_processing.py

Co-authored-by: Copilot <[email protected]>
Signed-off-by: flora-hofmann-frequenz <[email protected]>

Author: flora-hofmann-frequenz

src/frequenz/lib/notebooks/reporting/data_processing.py

@@ -105,7 +105,8 @@ def convert_timezone(df: pd.DataFrame) -> pd.DataFrame:
         A copy of the DataFrame with localized timezone.
     """
     df = df.copy()
-    assert COLUMN_TIMESTAMP in df.columns, df
+    if COLUMN_TIMESTAMP not in df.columns:
+        raise KeyError(f"Missing required column '{COLUMN_TIMESTAMP}' in DataFrame.")
     if df[COLUMN_TIMESTAMP].dt.tz is None:
         df[COLUMN_TIMESTAMP] = df[COLUMN_TIMESTAMP].dt.tz_localize("UTC")
     df[COLUMN_TIMESTAMP] = df[COLUMN_TIMESTAMP].dt.tz_convert(TZ_NAME)