numpy · charris · Sep 4, 2023 · Aug 31, 2023 · Sep 1, 2023 · Sep 1, 2023
diff --git a/environment.yml b/environment.yml
@@ -26,7 +26,7 @@ dependencies:
   - hypothesis
   # For type annotations
   - typing_extensions>=4.2.0  # needed for python < 3.10
-  - mypy=1.4.1
+  - mypy=1.5.1
   # For building docs
   - sphinx>=4.5.0
   - sphinx-design

diff --git a/numpy/random/mtrand.pyi b/numpy/random/mtrand.pyi
@@ -1,3 +1,4 @@
+import builtins
 from collections.abc import Callable
 from typing import Any, Union, overload, Literal
 
@@ -224,7 +225,7 @@ class RandomState:
         size: None | _ShapeLike = ...,
         dtype: dtype[uint] | type[uint] | _UIntCodes | _SupportsDType[dtype[uint]] = ...,
     ) -> ndarray[Any, dtype[uint]]: ...
-    def bytes(self, length: int) -> bytes: ...
+    def bytes(self, length: int) -> builtins.bytes: ...
     @overload
     def choice(
         self,

diff --git a/numpy/typing/mypy_plugin.py b/numpy/typing/mypy_plugin.py
@@ -75,8 +75,7 @@ def _get_precision_dict() -> dict[str, str]:
 
 
 def _get_extended_precision_list() -> list[str]:
-    extended_types = [np.ulonglong, np.longlong, np.longdouble, np.clongdouble]
-    extended_names = {
+    extended_names = [
         "uint128",
         "uint256",
         "int128",
@@ -89,8 +88,8 @@ def _get_extended_precision_list() -> list[str]:
         "complex192",
         "complex256",
         "complex512",
-    }
-    return [i.__name__ for i in extended_types if i.__name__ in extended_names]
+    ]
+    return [i for i in extended_names if hasattr(np, i)]
 
 
 def _get_c_intp_name() -> str:

diff --git a/numpy/typing/tests/data/fail/lib_polynomial.pyi b/numpy/typing/tests/data/fail/lib_polynomial.pyi
@@ -8,6 +8,11 @@ AR_U: npt.NDArray[np.str_]
 
 poly_obj: np.poly1d
 
+np.polymul(AR_f8, AR_U)  # E: incompatible type
+np.polydiv(AR_f8, AR_U)  # E: incompatible type
+
+5**poly_obj  # E: No overload variant
+
 np.polyint(AR_U)  # E: incompatible type
 np.polyint(AR_f8, m=1j)  # E: No overload variant
 
@@ -22,8 +27,3 @@ np.polyfit(AR_f8, AR_f8, 1, cov="bob")  # E: No overload variant
 np.polyval(AR_f8, AR_U)  # E: incompatible type
 np.polyadd(AR_f8, AR_U)  # E: incompatible type
 np.polysub(AR_f8, AR_U)  # E: incompatible type
-np.polymul(AR_f8, AR_U)  # E: incompatible type
-np.polydiv(AR_f8, AR_U)  # E: incompatible type
-
-5**poly_obj  # E: No overload variant
-    hash(poly_obj)
diff --git a/numpy/typing/tests/data/misc/extended_precision.pyi b/numpy/typing/tests/data/misc/extended_precision.pyi
@@ -1,17 +1,25 @@
+import sys
+
 import numpy as np
+from numpy._typing import _80Bit, _96Bit, _128Bit, _256Bit
+
+if sys.version_info >= (3, 11):
+    from typing import assert_type
+else:
+    from typing_extensions import assert_type
 
-reveal_type(np.uint128())
-reveal_type(np.uint256())
+assert_type(np.uint128(), np.unsignedinteger[_128Bit])
+assert_type(np.uint256(), np.unsignedinteger[_256Bit])
 
-reveal_type(np.int128())
-reveal_type(np.int256())
+assert_type(np.int128(), np.signedinteger[_128Bit])
+assert_type(np.int256(), np.signedinteger[_256Bit])
 
-reveal_type(np.float80())
-reveal_type(np.float96())
-reveal_type(np.float128())
-reveal_type(np.float256())
+assert_type(np.float80(), np.floating[_80Bit])
+assert_type(np.float96(), np.floating[_96Bit])
+assert_type(np.float128(), np.floating[_128Bit])
+assert_type(np.float256(), np.floating[_256Bit])
 
-reveal_type(np.complex160())
-reveal_type(np.complex192())
-reveal_type(np.complex256())
-reveal_type(np.complex512())
+assert_type(np.complex160(), np.complexfloating[_80Bit, _80Bit])
+assert_type(np.complex192(), np.complexfloating[_96Bit, _96Bit])
+assert_type(np.complex256(), np.complexfloating[_128Bit, _128Bit])
+assert_type(np.complex512(), np.complexfloating[_256Bit, _256Bit])
diff --git a/numpy/typing/tests/data/mypy.ini b/numpy/typing/tests/data/mypy.ini
@@ -2,3 +2,4 @@
 plugins = numpy.typing.mypy_plugin
 show_absolute_path = True
 implicit_reexport = False
+pretty = True