Fix annotations of CallList

getsentry · Oct 12, 2022 · 046e706 · 046e706
1 parent 27cb0c7
commit 046e706
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/CHANGES b/CHANGES
@@ -1,3 +1,8 @@
+0.23.0
+------
+
+* Fix type annotations of `CallList`.
+
 0.22.0
 ------
 

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]