🐛 Fixed wrong variable in rate limiter init

This commit is contained in:
GeoffreyCoulaud
2023-06-03 16:35:25 +02:00
parent 58054f1c26
commit b1476a744d

View File

@@ -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()