Skip to content

Commit

Permalink
[core][compiled graphs] Add ray[cg] pip install option (ray-project#4…
Browse files Browse the repository at this point in the history
…9220)

Signed-off-by: Rui Qiao <ruisearch42@gmail.com>
Signed-off-by: ujjawal-khare <ujjawal.khare@dream11.com>
  • Loading branch information
ruisearch42 authored and ujjawal-khare committed Dec 17, 2024
1 parent 294a9da commit 22c62e6
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion python/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -232,7 +232,7 @@ def get_packages(self):
"pyarrow <18; sys_platform == 'darwin' and platform_machine == 'x86_64'",
]
setup_spec.extras = {
"adag": [
"cg": [
"cupy-cuda12x; sys_platform != 'darwin'",
],
"client": [
Expand Down Expand Up @@ -287,6 +287,10 @@ def get_packages(self):
],
}

# Both "adag" and "cg" are for Compiled Graphs.
# "adag" is deprecated and will be removed in the future.
setup_spec.extras["adag"] = list(setup_spec.extras["cg"])

# Ray Serve depends on the Ray dashboard components.
setup_spec.extras["serve"] = list(
set(setup_spec.extras["serve"] + setup_spec.extras["default"])
Expand Down

0 comments on commit 22c62e6

Please sign in to comment.