mirror of
https://github.com/wassname/ray.git
synced 2026-08-13 12:30:18 +08:00
Improved KeyboardInterrupt Exception Handling (#5237)
This commit is contained in:
@@ -36,6 +36,7 @@ from ray.includes.unique_ids cimport (
|
||||
)
|
||||
from ray.includes.task cimport CTaskSpec
|
||||
from ray.includes.ray_config cimport RayConfig
|
||||
from ray.exceptions import RayletError
|
||||
from ray.utils import decode
|
||||
|
||||
cimport cpython
|
||||
@@ -57,7 +58,7 @@ cdef int check_status(const CRayStatus& status) nogil except -1:
|
||||
|
||||
with gil:
|
||||
message = status.message().decode()
|
||||
raise Exception(message)
|
||||
raise RayletError(message)
|
||||
|
||||
|
||||
cdef c_vector[CObjectID] ObjectIDsToVector(object_ids):
|
||||
|
||||
Reference in New Issue
Block a user