diff --git a/python/setup.py b/python/setup.py index 979551b971e61..550373d5dc8f5 100644 --- a/python/setup.py +++ b/python/setup.py @@ -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": [ @@ -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"])