diff --git a/novus/api/api_iterator.py b/novus/api/api_iterator.py index bb26ecf5..c2e7f871 100644 --- a/novus/api/api_iterator.py +++ b/novus/api/api_iterator.py @@ -66,10 +66,11 @@ async def __aiter__(self) -> AsyncGenerator[T, Any]: while self._remaining: # Work out what our request limit is - if isinstance(self._remaining, int): - limit = min(self.method_limit, self._remaining) - else: + if self._remaining is True: limit = self.method_limit + else: + # is an int + limit = min(self.method_limit, self._remaining) # Get items from the api items: list[T] = await self.method( @@ -84,7 +85,7 @@ async def __aiter__(self) -> AsyncGenerator[T, Any]: # Yield our just-given items for i in items: yield i - if isinstance(self._remaining, int): + if self._remaining is not True: self._remaining -= 1 # Break out of this if we didn't get the right number of messages