Fix __getitem__ signatures

gurobipy-stubs/gurobipy.pyi

@@ -29,6 +29,7 @@ _LinearComponent = Union[Var, LinExpr, MVar, MLinExpr]
 _QuadComponent = Union[QuadExpr, MQuadExpr]
 _ModelComponent = Union[Var, MVar, Constr, MConstr, QConstr, SOS, GenConstr]
 _ShapeLike = Union[int, Tuple[int, ...]]
+_IndexLike = Union[int, slice, Tuple[int, ...], Sequence[int], Tuple[Union[Sequence[int], int, slice], ...]]
 
 def abs_(__var: Var) -> GenExprAbs: ...
 @overload
@@ -865,6 +866,7 @@ class MLinExpr:
         self,
         other: np.ndarray
     ) -> TempMConstr: ...
+    def __getitem__(self, obj: _IndexLike) -> MLinExpr: ...
     @overload
     def __iadd__(self, other: _ConstComponent) -> MLinExpr: ...
     @overload
@@ -965,6 +967,7 @@ class MQuadExpr:
     def __ge__(self, other: _LinearComponent) -> TempMQConstr: ...
     @overload
     def __ge__(self, other: _QuadComponent) -> TempMQConstr: ...
+    def __getitem__(self, obj: _IndexLike) -> MQuadExpr: ...
     @overload
     def __iadd__(self, other: _ConstComponent) -> MQuadExpr: ...
     @overload
@@ -1096,22 +1099,7 @@ class MVar:
         self,
         other: np.ndarray
     ) -> TempMConstr: ...
-    @overload
-    def __getitem__(self, obj: int) -> Var: ...
-    # Cannot write concisely without overlapping signature 5
-    @overload
-    def __getitem__(  # type: ignore[misc]
-        self, obj: Tuple[int, ...]
-    ) -> Var: ...
-    @overload
-    def __getitem__(self, obj: slice) -> MVar: ...
-    @overload
-    def __getitem__(self, obj: Sequence[int]) -> MVar: ...
-    @overload
-    def __getitem__(
-        self,
-        obj: Tuple[Union[Sequence[int], int, slice], ...]
-    ) -> MVar: ...
+    def __getitem__(self, obj: _IndexLike) -> MVar: ...
     def __itruediv__(self, other: _ConstComponent) -> MLinExpr: ...
     @overload
     def __le__(self, other: Union[float, int]) -> TempMConstr: ...