fix: import + tests

sylvainmouquet · Sep 24, 2024 · fec99ee · fec99ee
1 parent 1e37038
commit fec99ee
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 8 deletions.
diff --git a/reattempt/__init__.py b/reattempt/__init__.py
@@ -1,5 +1,5 @@
 __version__ = "1.0.7"
-__all__ = ["reattempt"]
+__all__ = ("__version__", "reattempt")
 
 import asyncio
 import contextlib
@@ -17,9 +17,6 @@
 logger.addHandler(logging.NullHandler())
 
 
-def __call__(func, max_retries=CONST_DEFAULT_MAX_RETRIES, min_time=CONST_DEFAULT_MIN_TIME, max_time=CONST_DEFAULT_MAX_TIME):
-    return reattempt(func, max_retries, min_time, max_time)
-
 def reattempt(
     func=None,
     max_retries: int = CONST_DEFAULT_MAX_RETRIES,
@@ -199,5 +196,3 @@ async def retry_async_gen_func(*args, **kwargs):
         return decorator(func)
     return decorator
 
-import sys
-sys.modules[__name__] = reattempt
diff --git a/test/test_reattempt.py b/test/test_reattempt.py
@@ -1,7 +1,7 @@
 from typing import AsyncGenerator
 
 import pytest
-import reattempt
+from reattempt import reattempt
 from test.conftest import MAX_ATTEMPTS, MIN_TIME, MAX_TIME, RetryException
 
 

diff --git a/uv.lock b/uv.lock