Improve datetime conversion handling by adding fallback for ValueErro…

…r in convert_to_datetime function
carbonplan · Dec 3, 2024 · d33d753 · d33d753
1 parent 3d5b43b
commit d33d753
Showing 1 changed file with 5 additions and 3 deletions.
diff --git 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