Removes int support for dates

Author: spjuhel

climada/trajectories/snapshot.py

@@ -49,8 +49,8 @@ class Snapshot:
     exposure : Exposures
     hazard : Hazard
     impfset : ImpactFuncSet
-    date : int | datetime.date | str | pd.Timestamp
-        The date of the Snapshot, it can be an integer representing a year,
+    date : datetime.date | str | pd.Timestamp
+        The date of the Snapshot, it can be an string representing a year,
         a datetime object or a string representation of a datetime object.
     ref_only : bool, default False
         Should the `Snapshot` contain deep copies of the Exposures, Hazard and Impfset (False)
@@ -83,7 +83,7 @@ def __init__(
         hazard: Hazard,
         impfset: ImpactFuncSet,
         measure: Measure | None,
-        date: int | datetime.date | str | pd.Timestamp,
+        date: datetime.date | str | pd.Timestamp,
         ref_only: bool = False,
         _from_factory: bool = False,
     ) -> None:
@@ -107,7 +107,7 @@ def from_triplet(
         exposure: Exposures,
         hazard: Hazard,
         impfset: ImpactFuncSet,
-        date: int | datetime.date | str | pd.Timestamp,
+        date: datetime.date | str | pd.Timestamp,
         ref_only: bool = False,
     ) -> "Snapshot":
         """Create a Snapshot from exposure, hazard and impact functions set
@@ -122,7 +122,7 @@ def from_triplet(
         exposure : Exposures
         hazard : Hazard
         impfset : ImpactFuncSet
-        date : int | datetime.date | str | pd.Timestamp
+        date : datetime.date | str | pd.Timestamp
         ref_only : bool
             If true, uses references to the exposure, hazard and impact
             function objects. Note that modifying the original objects after
@@ -186,10 +186,7 @@ def impact_calc_data(self) -> dict:
 
     @staticmethod
     def _convert_to_timestamp(date_arg) -> pd.Timestamp:
-        """Convert date argument of type int or str or datetime.date to pandas Timestamp object."""
-        if isinstance(date_arg, int):
-            # Assume the integer represents a year
-            return pd.Timestamp(year=date_arg, month=1, day=1)
+        """Convert date argument of type str or datetime.date to pandas Timestamp object."""
         if isinstance(date_arg, str):
             # Try to parse the string as a date
             try:
@@ -201,9 +198,7 @@ def _convert_to_timestamp(date_arg) -> pd.Timestamp:
         if isinstance(date_arg, pd.Timestamp):
             return date_arg
 
-        raise TypeError(
-            "date_arg must be an int, str, datetime.date or pandas.Timestamp"
-        )
+        raise TypeError("date_arg must be an str, datetime.date or pandas.Timestamp")
 
     def apply_measure(self, measure: Measure) -> "Snapshot":
         """Create a new snapshot by applying a Measure object.

climada/trajectories/test/test_snapshot.py

@@ -57,7 +57,7 @@ def mock_context(shared_data):
         "mod_exp": modified_exp,
         "mod_haz": modified_haz,
         "mod_imp": modified_imp,
-        "date": pd.Timestamp(2023),
+        "date": pd.Timestamp("2023"),
     }
 
 
@@ -72,14 +72,14 @@ def test_not_from_factory_warning(mock_context):
             hazard=mock_context["haz"],
             impfset=mock_context["imp"],
             measure=None,
-            date=2001,
+            date="2001",
         )
 
 
 @pytest.mark.parametrize(
     "input_date,expected",
     [
-        (2023, pd.Timestamp(2023, 1, 1)),
+        ("2023", pd.Timestamp(2023, 1, 1)),
         ("2023-01-01", pd.Timestamp(2023, 1, 1)),
         (datetime.date(2023, 1, 1), pd.Timestamp(2023, 1, 1)),
     ],
@@ -108,9 +108,14 @@ def test_init_invalid_date_format(mock_context):
 def test_init_invalid_date_type(mock_context):
     with pytest.raises(
         TypeError,
-        match=r"date_arg must be an int, str, datetime.date or pandas.Timestamp",
+        match=r"date_arg must be an str, datetime.date or pandas.Timestamp",
     ):
-        Snapshot.from_triplet(exposure=mock_context["exp"], hazard=mock_context["haz"], impfset=mock_context["imp"], date=2023.5)  # type: ignore
+        Snapshot.from_triplet(
+            exposure=mock_context["exp"],
+            hazard=mock_context["haz"],
+            impfset=mock_context["imp"],
+            date=2023.5,
+        )  # type: ignore
 
 
 def test_properties(mock_context):