fix: TimeoutException maps to TimeoutException

automl · Mar 19, 2023 · 2470f56 · 2470f56
1 parent 2c0879d
commit 2470f56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pynisher/pynisher.py b/pynisher/pynisher.py
@@ -44,7 +44,7 @@ class Pynisher(Generic[P, T]):
     CpuTimeoutException = CpuTimeoutException
     MemoryLimitException = MemoryLimitException
     PynisherException = PynisherException
-    TimeoutException = WallTimeoutException
+    TimeoutException = TimeoutException
 
     # If `raises=True` or left as default, the return type when calling is just T
     @overload