[🔨] Fix bug in sync ratelimit decorator

WizzyGeek · Nov 22, 2021 · 4c388c4 · 4c388c4
1 parent 308d671
commit 4c388c4
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,7 +1,7 @@
 
 [tool.poetry]
 name = "uprate"
-version = "0.2.0"
+version = "0.2.1"
 description = "Ratelimits. Dumbified. A fully typed, simple ratelimit library."
 authors = ["WizzyGeek <[email protected]>"]
 license = "MIT"

diff --git a/uprate/_sync.py b/uprate/_sync.py
@@ -15,6 +15,12 @@
 from .rate import Rate, RateGroup
 from .store import H, MemoryStore, T
 
+__all__ = (
+ "SyncStore",
+ "SyncMemoryStore",
+ "SyncRateLimit"
+)
+
 G = TypeVar("G")
 
 if TYPE_CHECKING:

diff --git a/uprate/_utils.py b/uprate/_utils.py
@@ -6,7 +6,7 @@
 R = TypeVar("R")
 
 __all__ = (
- "maybe_awaitable"
+ "maybe_awaitable",
 )
 
 async def maybe_awaitable(ret: Union[R, Awaitable[R]]) -> R:

diff --git a/uprate/decorators.py b/uprate/decorators.py
@@ -138,10 +138,11 @@ def rated(*args, **kwargs):
  if on_retry is None:
  raise err from None
  else:
- on_retry()
+ on_retry(err)
  else:
  return func()
 
  return _apply_attrs(rated, limit=limit)
 
  return decorator
+