From bc532aafc5fc3a9c3bfffad1a5ba8a2851338a6b Mon Sep 17 00:00:00 2001 From: Sahil Jhawar Date: Thu, 9 Jul 2026 11:54:44 +0200 Subject: [PATCH] chore: fix wdc time assertion --- swvo/io/dst/wdc.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/swvo/io/dst/wdc.py b/swvo/io/dst/wdc.py index 1e2a2aa..5509635 100644 --- a/swvo/io/dst/wdc.py +++ b/swvo/io/dst/wdc.py @@ -66,7 +66,10 @@ def download_and_process(self, start_time: datetime, end_time: datetime, reproce None """ - assert start_time < end_time, "Start time must be before end time" + start_time = enforce_utc_timezone(start_time) + end_time = enforce_utc_timezone(end_time) + + assert start_time < end_time, "Start time must be before end time!" temporary_dir = Path("./temp_wdc") temporary_dir.mkdir(exist_ok=True, parents=True) @@ -225,11 +228,6 @@ def read(self, start_time: datetime, end_time: datetime, download: bool = False) WDC Dst data. """ - if start_time > end_time: - msg = "start_time must be before end_time" - logger.error(msg) - raise ValueError(msg) - start_time = enforce_utc_timezone(start_time) end_time = enforce_utc_timezone(end_time)