Handle ValueError in date parsing to improve robustness of set_vcs_vi…

…ntage_year function (#90)
carbonplan · Nov 21, 2024 · d6e29e2 · d6e29e2
1 parent 9dd0602
commit d6e29e2
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/offsets_db_data/vcs.py b/offsets_db_data/vcs.py
@@ -109,7 +109,10 @@ def set_vcs_vintage_year(df: pd.DataFrame, *, date_column: str) -> pd.DataFrame:
         DataFrame with a new 'vintage' column, containing the vintage year of each transaction.
     """
 
-    df[date_column] = pd.to_datetime(df[date_column], format='%d/%m/%Y', utc=True)
+    try:
+        df[date_column] = pd.to_datetime(df[date_column], format='%d/%m/%Y', utc=True)
+    except ValueError:
+        df[date_column] = pd.to_datetime(df[date_column], utc=True)
     df['vintage'] = df[date_column].dt.year
     return df