Fix annotations of CallList

getsentry · Oct 12, 2022 · b958e84 · b958e84
1 parent 27cb0c7
commit b958e84
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/responses/__init__.py b/responses/__init__.py
@@ -230,6 +230,14 @@ def __iter__(self) -> Iterator[Call]:
     def __len__(self) -> int:
         return len(self._calls)
 
+    @overload
+    def __getitem__(self, idx: int) -> Call:
+        ...
+
+    @overload
+    def __getitem__(self, idx: slice) -> List[Call]:
+        ...
+
     def __getitem__(self, idx: Union[int, slice]) -> Union[Call, List[Call]]:
         return self._calls[idx]