.

spacemanspiff2007 · Dec 6, 2024 · 699435c · 699435c
1 parent e47d5bb
commit 699435c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/HABApp/util/rate_limiter/limits/fixed_window.py b/src/HABApp/util/rate_limiter/limits/fixed_window.py
@@ -1,6 +1,7 @@
 from dataclasses import dataclass
 from time import monotonic
-from typing import override
+
+from typing_extensions import override
 
 from .base import BaseRateLimit, BaseRateLimitInfo
 

diff --git a/src/HABApp/util/rate_limiter/limits/leaky_bucket.py b/src/HABApp/util/rate_limiter/limits/leaky_bucket.py
@@ -1,11 +1,11 @@
-from dataclasses import dataclass
 from time import monotonic
-from typing import Final, override
+from typing import Final
+
+from typing_extensions import override
 
 from .base import BaseRateLimit, BaseRateLimitInfo
 
 
-@dataclass
 class LeakyBucketLimitInfo(BaseRateLimitInfo):
     time_remaining: float  #: Time remaining until the next drop