From d33d753aa04201a20b9df2b454a7b8bb66dbf5e1 Mon Sep 17 00:00:00 2001 From: Anderson Banihirwe Date: Mon, 2 Dec 2024 21:46:47 -0800 Subject: [PATCH] Improve datetime conversion handling by adding fallback for ValueError in convert_to_datetime function --- offsets_db_data/common.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/offsets_db_data/common.py b/offsets_db_data/common.py index 01e03b6..da7ec71 100644 --- a/offsets_db_data/common.py +++ b/offsets_db_data/common.py @@ -100,10 +100,12 @@ def convert_to_datetime( """ for column in columns: - if column in df.columns: - df[column] = pd.to_datetime(df[column], utc=utc, **kwargs).dt.normalize() - else: + if column not in df.columns: raise KeyError(f"The column '{column}' is missing.") + try: + df[column] = pd.to_datetime(df[column], utc=utc, **kwargs).dt.normalize() + except ValueError: + df[column] = pd.to_datetime(df[column], utc=utc).dt.normalize() return df