From 128909377ae6c0a918f74a87894f404d058e92a8 Mon Sep 17 00:00:00 2001 From: Frederik Kratzert Date: Tue, 21 Oct 2025 14:54:38 +0000 Subject: [PATCH] Fix some warnings in Brazil fetcher --- examples/test_brazil_fetcher.py | 8 ++++---- rivretrieve/brazil.py | 7 ++++++- 2 files changed, 10 insertions(+), 5 deletions(-) diff --git a/examples/test_brazil_fetcher.py b/examples/test_brazil_fetcher.py index 08e55fa..8aa4ad0 100644 --- a/examples/test_brazil_fetcher.py +++ b/examples/test_brazil_fetcher.py @@ -7,11 +7,11 @@ logging.basicConfig(level=logging.INFO) gauge_ids = [ - "15400000", + "14515000", ] -variable = constants.STAGE_DAILY_MEAN -start_date = "2023-01-01" -end_date = "2024-12-31" +variable = constants.DISCHARGE_DAILY_MEAN +start_date = "1980-01-01" +end_date = None plt.figure(figsize=(12, 6)) diff --git a/rivretrieve/brazil.py b/rivretrieve/brazil.py index d2383fd..fa8146a 100644 --- a/rivretrieve/brazil.py +++ b/rivretrieve/brazil.py @@ -305,7 +305,8 @@ def _parse_data(self, gauge_id: str, raw_data: List[Dict[str, Any]], variable: s continue month_df = pd.DataFrame(list(day_values.items()), columns=[constants.TIME_INDEX, variable]) - all_dfs.append(month_df) + if not month_df.empty: + all_dfs.append(month_df) if not all_dfs: return pd.DataFrame(columns=[constants.TIME_INDEX, variable]) @@ -361,6 +362,10 @@ def get_data( try: raw_data = self._download_data(gauge_id, variable, start_date, end_date) df = self._parse_data(gauge_id, raw_data, variable) + + if df.empty: + return df + start_date_dt = pd.to_datetime(start_date) end_date_dt = pd.to_datetime(end_date) df = df[(df.index >= start_date_dt) & (df.index <= end_date_dt)]