precommit

Author: arthurlw

pandas/core/algorithms.py

@@ -24,6 +24,7 @@
     lib,
 )
 from pandas._libs.missing import NA
+from pandas._libs.tslibs.timestamps import Timestamp
 from pandas._typing import (
     AnyArrayLike,
     ArrayLike,
@@ -79,7 +80,6 @@
     na_value_for_dtype,
 )
 
-from pandas._libs.tslibs.timestamps import Timestamp
 from pandas.core.array_algos.take import take_nd
 from pandas.core.construction import (
     array as pd_array,
@@ -1695,7 +1695,7 @@ def map_array(
 
         if isinstance(arr.dtype, ArrowDtype) and arr.dtype.name.startswith("timestamp"):
             try:
-                # Convert elements to pandas.Timestamp (or datetime64[ns]) for dict lookup
+                # Convert elements to pandas.Timestamp (or datetime64[ns])
                 arr = arr.astype("datetime64[ns]")
             except Exception:
                 # fallback: safe, slow path

pandas/tests/series/methods/test_map.py

@@ -605,10 +605,11 @@ def test_map_kwargs():
     expected = Series([4, 6, 7])
     tm.assert_series_equal(result, expected)
 
+
 def test_map_arrow_timestamp_dict():
     # GH 61231
-    ser = pd.Series(pd.date_range("2023-01-01", periods=3)).astype("timestamp[ns][pyarrow]")
+    ser = Series(date_range("2023-01-01", periods=3)).astype("timestamp[ns][pyarrow]")
     mapper = {ts: i for i, ts in enumerate(ser)}
     result = ser.map(mapper)
-    expected = pd.Series([0, 1, 2], dtype="int64")
-    tm.assert_series_equal(result, expected)
+    expected = Series([0, 1, 2], dtype="int64")
+    tm.assert_series_equal(result, expected)