Datetime loading fix (#10)

* Fix dataset loading * Add catch if format isn't day-first
synthesized-io · Mar 13, 2024 · 773c556 · 773c556
1 parent 84e3c0b
commit 773c556
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/synthesized_datasets/_datasets.py b/src/synthesized_datasets/_datasets.py
@@ -87,7 +87,10 @@ def load(self) -> _pd.DataFrame:
             df = _pd.read_csv(self.url, dtype=dtypes)
             for col, dtype in self._schema.items():
                 if dtype in [_DType.DATETIME, _DType.DATE]:
-                    df[col] = _pd.to_datetime(df[col], dayfirst=True)
+                    try:
+                        df[col] = _pd.to_datetime(df[col])
+                    except ValueError:
+                        df[col] = _pd.to_datetime(df[col], dayfirst=True)
                 if dtype in [_DType.TIMEDELTA, _DType.TIME]:
                     df[col] = _pd.to_timedelta(df[col])
         df.attrs["name"] = self.name