Add timeout param to ray.get (#6107)

This commit is contained in:
Ujval Misra
2019-11-14 00:50:04 -08:00
committed by Richard Liaw
parent e4c0843f60
commit e3e3ad4b25
12 changed files with 100 additions and 20 deletions
+6
View File
@@ -161,6 +161,11 @@ class UnreconstructableError(RayError):
"https://ray.readthedocs.io/en/latest/memory-management.html"))
class RayTimeoutError(RayError):
"""Indicates that a call to the worker timed out."""
pass
RAY_EXCEPTION_TYPES = [
RayError,
RayTaskError,
@@ -168,4 +173,5 @@ RAY_EXCEPTION_TYPES = [
RayActorError,
ObjectStoreFullError,
UnreconstructableError,
RayTimeoutError,
]