[SPARK-48710][PYTHON] Fix type checks for NumPy in tests

Author: codesorcery

python/pyspark/core/rdd.py

@@ -5377,7 +5377,7 @@ def _test() -> None:
         from pandas.util.version import Version
 
         if Version(np.__version__) >= Version("2"):
-            np.set_printoptions(legacy="1.25")
+            np.set_printoptions(legacy="1.25")  # type: ignore[arg-type] # `legacy="1.25"` only available in `nump>=2`
     except TypeError:
         pass
 

python/pyspark/pandas/indexes/base.py

@@ -2652,7 +2652,7 @@ def _test() -> None:
     if Version(np.__version__) >= Version("2"):
         # Numpy 2.0+ changed its string format,
         # adding type information to numeric scalars.
-        np.set_printoptions(legacy="1.25")
+        np.set_printoptions(legacy="1.25")  # type: ignore[arg-type] # `legacy="1.25"` only available in `nump>=2`
 
     globs = pyspark.pandas.indexes.base.__dict__.copy()
     globs["ps"] = pyspark.pandas

python/pyspark/pandas/indexing.py

@@ -1840,7 +1840,7 @@ def _test() -> None:
     if Version(np.__version__) >= Version("2"):
         # Numpy 2.0+ changed its string format,
         # adding type information to numeric scalars.
-        np.set_printoptions(legacy="1.25")
+        np.set_printoptions(legacy="1.25")  # type: ignore[arg-type] # `legacy="1.25"` only available in `nump>=2`
 
     globs = pyspark.pandas.indexing.__dict__.copy()
     globs["ps"] = pyspark.pandas

python/pyspark/pandas/namespace.py

@@ -3738,7 +3738,7 @@ def _test() -> None:
     if Version(np.__version__) >= Version("2"):
         # Numpy 2.0+ changed its string format,
         # adding type information to numeric scalars.
-        np.set_printoptions(legacy="1.25")
+        np.set_printoptions(legacy="1.25")  # type: ignore[arg-type] # `legacy="1.25"` only available in `nump>=2`
 
     globs = pyspark.pandas.namespace.__dict__.copy()
     globs["ps"] = pyspark.pandas