fix(utils): Fix delay issue in Bucket.

2023-10-01 13:33:30 +03:00
parent 89d236f0cb
commit c8223725ad
1 changed files with 4 additions and 1 deletions
--- a/src/utils/ratelimits.py
+++ b/src/utils/ratelimits.py
@@ -56,7 +56,10 @@ class Bucket:
    def delay(self):
        self._leak()
        if self._level + 1 > self.max_level:
-            return (self._level + 1 - self.max_level) * self.leak_rate
+            delay = (self._level + 1 - self.max_level) * self.leak_rate
+        else:
+            delay = 0
+        return delay

    def _leak(self):
        if self._level: