chore: Fix mypy

CMIP-REF · Dec 9, 2024 · b36e493 · b36e493
1 parent 0f9d7d4
commit b36e493
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/packages/ref/src/ref/datasets/cmip6.py b/packages/ref/src/ref/datasets/cmip6.py
@@ -23,7 +23,7 @@ def _parse_datetime(dt_str: pd.Series[str]) -> pd.Series[datetime | Any]:
     Pandas tries to coerce everything to their own datetime format, which is not what we want here.
     """
 
-    def _inner(date_string):
+    def _inner(date_string: str | None) -> datetime | None:
         if not date_string:
             return None
 

diff --git a/packages/ref/tests/unit/datasets/test_cmip6.py b/packages/ref/tests/unit/datasets/test_cmip6.py
@@ -19,9 +19,9 @@ def check(df: pd.DataFrame, basename: str):
 
 def test_parse_datetime():
     pd.testing.assert_series_equal(
-        _parse_datetime(pd.Series(["2021-01-01 00:00:00", "1850-01-17 00:29:59.999993"])),
+        _parse_datetime(pd.Series(["2021-01-01 00:00:00", "1850-01-17 00:29:59.999993", None])),
         pd.Series(
-            [datetime.datetime(2021, 1, 1, 0, 0), datetime.datetime(1850, 1, 17, 0, 29, 59, 999993)],
+            [datetime.datetime(2021, 1, 1, 0, 0), datetime.datetime(1850, 1, 17, 0, 29, 59, 999993), None],
             dtype="object",
         ),
     )