diff --git a/src/token_bucket/limiter.py b/src/token_bucket/limiter.py index eb05f82..b3521b5 100644 --- a/src/token_bucket/limiter.py +++ b/src/token_bucket/limiter.py @@ -114,9 +114,11 @@ def consume(self, key: KeyType, num_tokens: int = 1) -> bool: return self._storage.consume(key, num_tokens) def rate(self) -> Union[float, int]: - # Return the token bucket rate, needed for use cases where you dynamically wait based on the remaining tokens + # Return the token bucket rate, needed for use cases where + # you dynamically wait based on the remaining tokens return self._rate def capacity(self) -> int: - # Return the token bucket rate, needed for use cases where you dynamically wait based on the remaining tokens + # Return the token bucket rate, needed for use cases where + # you dynamically wait based on the remaining tokens return self._capacity