feat: Add exception types to Pynisher class

automl · Mar 19, 2023 · 2c0879d · 2c0879d
1 parent 70f5a4c
commit 2c0879d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/pynisher/pynisher.py b/pynisher/pynisher.py
@@ -40,6 +40,12 @@ class _EMPTY:
 class Pynisher(Generic[P, T]):
     """Restrict a function's resources"""
 
+    WallTimeoutException = WallTimeoutException
+    CpuTimeoutException = CpuTimeoutException
+    MemoryLimitException = MemoryLimitException
+    PynisherException = PynisherException
+    TimeoutException = WallTimeoutException
+
     # If `raises=True` or left as default, the return type when calling is just T
     @overload
     def __init__(

diff --git a/setup.py b/setup.py
@@ -9,7 +9,7 @@
 
 setup(
     name="pynisher",
-    version="1.0.2",
+    version="1.0.3",
     packages=find_packages(where=".", include=["pynisher*"], exclude=["test*"]),
     include_package_data=True,
     install_requires=[