fix: Series.dt argument 'ambiguous'

Author: cmp0xff

pandas-stubs/core/indexes/accessors.pyi

@@ -167,7 +167,7 @@ class _DatetimeRoundingMethods(Generic[_DTTimestampTimedeltaReturnType]):
     def round(
         self,
         freq: str | BaseOffset | None,
-        ambiguous: Literal["raise", "infer", "NaT"] | np_ndarray_bool = ...,
+        ambiguous: Literal["raise", "infer", "NaT"] | bool | np_ndarray_bool = ...,
         nonexistent: (
             Literal["shift_forward", "shift_backward", "NaT", "raise"]
             | timedelta
@@ -177,7 +177,7 @@ class _DatetimeRoundingMethods(Generic[_DTTimestampTimedeltaReturnType]):
     def floor(
         self,
         freq: str | BaseOffset | None,
-        ambiguous: Literal["raise", "infer", "NaT"] | np_ndarray_bool = ...,
+        ambiguous: Literal["raise", "infer", "NaT"] | bool | np_ndarray_bool = ...,
         nonexistent: (
             Literal["shift_forward", "shift_backward", "NaT", "raise"]
             | timedelta
@@ -187,7 +187,7 @@ class _DatetimeRoundingMethods(Generic[_DTTimestampTimedeltaReturnType]):
     def ceil(
         self,
         freq: str | BaseOffset | None,
-        ambiguous: Literal["raise", "infer", "NaT"] | np_ndarray_bool = ...,
+        ambiguous: Literal["raise", "infer", "NaT"] | bool | np_ndarray_bool = ...,
         nonexistent: (
             Literal["shift_forward", "shift_backward", "NaT", "raise"]
             | timedelta

tests/test_io.py

@@ -43,7 +43,6 @@
 from pandas._testing import ensure_clean
 import pytest
 import sqlalchemy
-import sqlalchemy.ext.declarative
 import sqlalchemy.orm
 import sqlalchemy.orm.decl_api
 from typing_extensions import assert_type

tests/test_timefuncs.py

@@ -489,16 +489,31 @@ def test_series_dt_accessors() -> None:
         pd.Series,
         pd.Timestamp,
     )
+    check(
+        assert_type(s0.dt.round("D", ambiguous=False), "TimestampSeries"),
+        pd.Series,
+        pd.Timestamp,
+    )
     check(
         assert_type(s0.dt.floor("D", nonexistent=dt.timedelta(1)), "TimestampSeries"),
         pd.Series,
         pd.Timestamp,
     )
+    check(
+        assert_type(s0.dt.floor("D", ambiguous=False), "TimestampSeries"),
+        pd.Series,
+        pd.Timestamp,
+    )
     check(
         assert_type(s0.dt.ceil("D", nonexistent=dt.timedelta(1)), "TimestampSeries"),
         pd.Series,
         pd.Timestamp,
     )
+    check(
+        assert_type(s0.dt.ceil("D", ambiguous=False), "TimestampSeries"),
+        pd.Series,
+        pd.Timestamp,
+    )
     check(assert_type(s0.dt.month_name(), "pd.Series[str]"), pd.Series, str)
     check(assert_type(s0.dt.day_name(), "pd.Series[str]"), pd.Series, str)
     check(assert_type(s0.dt.unit, TimeUnit), str)