Fix errors on py310

Author: hamdanal

pandas-stubs/core/indexes/interval.pyi

@@ -222,7 +222,7 @@ class IntervalIndex(ExtensionIndex[IntervalT], IntervalMixin):
     def memory_usage(self, deep: bool = False) -> int: ...
     @property
     def is_overlapping(self) -> bool: ...
-    def get_loc(self, key: Label) -> int | slice | npt.NDArray[np.bool_]: ...
+    def get_loc(self, key: Label) -> int | slice | np_1darray[np.bool]: ...
     @final
     def get_indexer(
         self,

tests/test_indexes.py

@@ -1313,6 +1313,22 @@ def test_get_loc() -> None:
         np_1darray[np.bool],
     )
 
+    i1, i2, i3 = pd.Interval(0, 1), pd.Interval(1, 2), pd.Interval(0, 2)
+    unique_interval_index = pd.IntervalIndex([i1, i2])
+    check(
+        assert_type(
+            unique_interval_index.get_loc(i1), Union[int, slice, np_1darray[np.bool]]
+        ),
+        np.int64,
+    )
+    overlap_interval_index = pd.IntervalIndex([i1, i2, i3])
+    check(
+        assert_type(
+            overlap_interval_index.get_loc(1), Union[int, slice, np_1darray[np.bool]]
+        ),
+        np_1darray[np.bool],
+    )
+
 
 def test_value_counts() -> None:
     nmi = pd.Index(list("abcb"))

tests/test_scalars.py

@@ -1239,7 +1239,9 @@ def test_timestamp_cmp() -> None:
     np_dt64_arr: npt.NDArray[np.datetime64] = np.array(
         [1, 2, 3], dtype="datetime64[ns]"
     )
-    np_dt64_arr2d = np.arange(6).astype(dtype=np.datetime64).reshape(3, 2)
+    np_dt64_arr2d: np.ndarray[tuple[int, int], np.dtype[np.datetime64]] = (
+        np.arange(6).astype(dtype=np.datetime64).reshape(3, 2)
+    )
 
     c_timestamp = ts
     c_np_dt64 = np.datetime64(1, "ns")