Add class and decorator to __init__

Author: Doru Irimescu

src/exception_with_retry/__init__.py

@@ -6,6 +6,9 @@
 else:
     from importlib_metadata import PackageNotFoundError, version  # pragma: no cover
 
+from exception_with_retry.source import ExceptionWithRetry, exception_with_retry
+
+__all__ = ["ExceptionWithRetry", "exception_with_retry"]
 try:
     # Change here if project is renamed and does not equal the package name
     dist_name = "exception-with-retry"

tests/test_1.py

@@ -1,7 +1,7 @@
 import unittest
 from typing import List
 
-from exception_with_retry.source import ExceptionWithRetry, exception_with_retry
+from exception_with_retry import ExceptionWithRetry, exception_with_retry
 
 __author__ = "Doru Irimescu"
 __copyright__ = "Doru Irimescu"