diff --git a/pyproject.toml b/pyproject.toml index 0db76bc..a840b36 100644 --- 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 "] license = "MIT" diff --git a/uprate/_sync.py b/uprate/_sync.py index 51dd52d..ba2cd9b 100644 --- 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 index 5c6527a..1f5d2bc 100644 --- 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 index 8808480..a73ba40 100644 --- 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 +