From c476037c97ea0161dddf106a57bbf547ce5467f8 Mon Sep 17 00:00:00 2001
From: Simon Mo <simon.mo@hey.com>
Date: Tue, 17 Nov 2020 17:20:30 -0800
Subject: [PATCH] [Core] Async API should raise on all RayError (#12043)

Before this PR we are raising just RayTaskError, this means errors
like RayActorError(Actor Died) won't be propogated and thrown at
`await object_ref`. This PR fixes that.
---
 python/ray/_raylet.pyx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/ray/_raylet.pyx b/python/ray/_raylet.pyx
index 4561d9dc4..65d6ed6ff 100644
--- a/python/ray/_raylet.pyx
+++ b/python/ray/_raylet.pyx
@@ -1524,7 +1524,7 @@ cdef void async_set_result(shared_ptr[CRayObject] obj,
             cpython.Py_DECREF(py_future)
             return
 
-        if isinstance(result, RayTaskError):
+        if isinstance(result, RayError):
             ray.worker.last_task_error_raise_time = time.time()
             py_future.set_exception(result.as_instanceof_cause())
         else: