From b1476a744dd4b024796ab67c00a79e26af27126e Mon Sep 17 00:00:00 2001 From: GeoffreyCoulaud Date: Sat, 3 Jun 2023 16:35:25 +0200 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20Fixed=20wrong=20variable=20in=20?= =?UTF-8?q?rate=20limiter=20init?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/utils/rate_limiter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/utils/rate_limiter.py b/src/utils/rate_limiter.py index b90f93d..ca118b4 100644 --- a/src/utils/rate_limiter.py +++ b/src/utils/rate_limiter.py @@ -45,7 +45,7 @@ class TokenBucketRateLimiter(AbstractContextManager): # Initialize the bucket self.bucket = BoundedSemaphore(self.MAX_TOKENS) missing = 0 if initial_tokens is None else self.MAX_TOKENS - initial_tokens - missing = max(0, min(missing, max_tokens)) + missing = max(0, min(missing, self.MAX_TOKENS)) for _ in range(missing): self.bucket.acquire()