Made some changes:

- Removed stub and test for applymap
- Downgraded pyright
- Ran the precommit (it just removes some whitespace)

Author: amackinnon1

pandas-stubs/core/frame.pyi

@@ -1316,7 +1316,11 @@ class DataFrame(NDFrame, OpsMixin):
     ) -> DataFrame: ...
 
     # Add spacing between apply() overloads and remaining annotations
-    # In pandas 2.1.0 and later, arraymap deprecated replaced with map
+    # Deprecated in pandas 2.1.0 and later
+    def applymap(
+        self, func: Callable, na_action: Literal["ignore"] | None = ..., **kwargs
+    ) -> DataFrame: ...
+    # In pandas 2.1.0 and later: applymap is renamed map, applymap is deprecated
     def map(
         self, func: Callable, na_action: Literal["ignore"] | None = ..., **kwargs
     ) -> DataFrame: ...

pyproject.toml

@@ -39,7 +39,7 @@ mypy = "1.6.0"
 pandas = "2.1.1"
 pyarrow = ">=10.0.1"
 pytest = ">=7.1.2"
-pyright = "==1.1.331"  # Don't use 1.1.332
+pyright = ">=1.1.331"
 poethepoet = ">=0.16.5"
 loguru = ">=0.6.0"
 typing-extensions = ">=4.4.0"

tests/test_frame.py

@@ -728,7 +728,21 @@ def gethead(s: pd.Series, y: int) -> pd.Series:
     )
 
 
-# In pandas 2.1.0 and later, arraymap deprecated replaced with map
+# Deprecated in pandas 2.1.0
+def test_types_applymap() -> None:
+    with pytest_warns_bounded(
+        FutureWarning, "DataFrame.applymap has been deprecated", lower="2.0.99"
+    ):
+        df = pd.DataFrame(data={"col1": [2, 1], "col2": [3, 4]})
+        df.applymap(lambda x: x**2)
+        df.applymap(np.exp)
+        df.applymap(str)
+        # na_action parameter was added in 1.2.0 https://pandas.pydata.org/docs/whatsnew/v1.2.0.html
+        df.applymap(np.exp, na_action="ignore")
+        df.applymap(str, na_action=None)
+
+
+# Added in pandas 2.1.0
 def test_types_map() -> None:
     df = pd.DataFrame(data={"col1": [2, 1], "col2": [3, 4]})
     df.map(lambda x: x**2)