updates to ray CLI

beacon-biosignals · Oct 26, 2023 · 0b249d3 · 0b249d3
1 parent 9cd5715
commit 0b249d3
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 2 deletions.
diff --git a/build/BUILD.bazel b/build/BUILD.bazel
@@ -28,6 +28,7 @@ cc_binary(
         "@com_github_ray_project_ray//:global_state_accessor_lib",
         "@julia//:headers",
         "@libcxxwrap_julia//:headers",
+        "@com_github_ray_project_ray//:gcs_server_lib",
     ],
     linkshared=True,
 )
diff --git a/build/WORKSPACE.bazel.tpl b/build/WORKSPACE.bazel.tpl
@@ -87,3 +87,16 @@ ray_deps_build_all()
 load("@com_github_grpc_grpc//bazel:grpc_extra_deps.bzl", "grpc_extra_deps")
 
 grpc_extra_deps()
+
+load("@rules_python//python:repositories.bzl", "python_register_toolchains")
+
+python_register_toolchains(
+    name = "python3_9",
+    python_version = "3.9",
+    register_toolchains = False,
+)
+
+load("@python3_9//:defs.bzl", python39 = "interpreter")
+load("@rules_python//python/pip_install:repositories.bzl", "pip_install_dependencies")
+
+pip_install_dependencies()
diff --git a/build/ray_commit b/build/ray_commit
@@ -1 +1 @@
-448a83caf44108fc1bc44fa7c6c358cffcfcb0d7
+8e3a05bd3affa23af965e5fa30c28ec4ce8e1eb5
diff --git a/build/wrapper.cc b/build/wrapper.cc
@@ -81,6 +81,7 @@ void initialize_worker(
     options.metrics_agent_port = -1;
     options.startup_token = startup_token;
     options.runtime_env_hash = runtime_env_hash;
+    // https://github.com/ray-project/ray/blob/4ab0ba0823a9d113a6a0ceb10d7ddd56596e9c1a/src/ray/core_worker/test/mock_worker.cc#L52
     options.task_execution_callback =
         [task_executor](
             const rpc::Address &caller_address,
@@ -94,13 +95,15 @@ void initialize_worker(
             const std::string &serialized_retry_exception_allowlist,
             std::vector<std::pair<ObjectID, std::shared_ptr<RayObject>>> *returns,
             std::vector<std::pair<ObjectID, std::shared_ptr<RayObject>>> *dynamic_returns,
+            std::vector<std::pair<ObjectID, bool>> *streaming_generator_returns,
             std::shared_ptr<LocalMemoryBuffer> &creation_task_exception_pb_bytes,
             bool *is_retryable_error,
             std::string *application_error,
             const std::vector<ConcurrencyGroup> &defined_concurrency_groups,
             const std::string name_of_concurrency_group_to_execute,
             bool is_reattempt,
-            bool is_streaming_generator) {
+            bool is_streaming_generator,
+            bool should_retry_exceptions) {
 
           std::vector<std::shared_ptr<RayObject>> return_vec;
           task_executor(ray_function,
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		448a83caf44108fc1bc44fa7c6c358cffcfcb0d7
		8e3a05bd3affa23af965e5fa30c28ec4ce8e1eb5