mirror of
https://github.com/wassname/ray.git
synced 2026-07-26 13:37:24 +08:00
[direct task] Retry tasks on failure and turn on RAY_FORCE_DIRECT for test_multinode_failures.py (#6306)
* multinode failures direct * Add number of retries allowed for tasks * Retry tasks * Add failing test for object reconstruction * Handle return status and debug * update * Retry task unit test * update * update * todo * Fix max_retries decorator, fix test * Fix test that flaked * lint * comments
This commit is contained in:
@@ -911,7 +911,8 @@ cdef class CoreWorker:
|
||||
args,
|
||||
int num_return_vals,
|
||||
c_bool is_direct_call,
|
||||
resources):
|
||||
resources,
|
||||
int max_retries):
|
||||
cdef:
|
||||
unordered_map[c_string, double] c_resources
|
||||
CTaskOptions task_options
|
||||
@@ -929,7 +930,8 @@ cdef class CoreWorker:
|
||||
|
||||
with nogil:
|
||||
check_status(self.core_worker.get().SubmitTask(
|
||||
ray_function, args_vector, task_options, &return_ids))
|
||||
ray_function, args_vector, task_options, &return_ids,
|
||||
max_retries))
|
||||
|
||||
return VectorToObjectIDs(return_ids)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user