updated according to reviewer suggestions

Author: arthurlw

pandas/core/arrays/arrow/array.py

@@ -1481,10 +1481,10 @@ def to_numpy(
         return result
 
     def map(self, mapper, na_action: Literal["ignore"] | None = None):
-        if is_numeric_dtype(self.dtype):
+        if is_numeric_dtype(self.dtype) or self.dtype == "timestamp[ns][pyarrow]":
             return map_array(self.to_numpy(), mapper, na_action=na_action)
-        elif self.dtype == "timestamp[ns][pyarrow]":
-            return map_array(self.to_numpy(dtype=object), mapper, na_action=na_action)
+        # elif self.dtype == "timestamp[ns][pyarrow]":
+        #     return map_array(self.to_numpy(), mapper, na_action=na_action)
         else:
             return super().map(mapper, na_action)
 

pandas/tests/series/methods/test_map.py

@@ -608,7 +608,7 @@ def test_map_kwargs():
 
 def test_map_arrow_timestamp_dict():
     # GH 61231
-    pytest.importorskip("pyarrow", minversion="10.0.1")
+    pytest.importorskip("pyarrow")
 
     ser = Series(date_range("2023-01-01", periods=3)).astype("timestamp[ns][pyarrow]")
     mapper = {ts: i for i, ts in enumerate(ser)}