[Build][Deps] Remove azure packages from core dependency

Closes: ray-project#48733 Signed-off-by: Chi-Sheng Liu <chishengliu@chishengliu.com>
MortalHappiness · Nov 25, 2024 · 34c99f7 · 34c99f7
1 parent ba8674a
commit 34c99f7
Show file tree

Hide file tree

Showing 5 changed files with 15 additions and 3 deletions.
diff --git a/python/ray/autoscaler/azure/defaults.yaml b/python/ray/autoscaler/azure/defaults.yaml
@@ -136,7 +136,7 @@ setup_commands:
 
 # Custom commands that will be run on the head node after common setup.
 head_setup_commands:
-    - pip install -U azure-cli-core==2.29.1 azure-identity==1.7.0 azure-mgmt-compute==23.1.0 azure-mgmt-network==19.0.0 azure-mgmt-resource==20.0.0 msrestazure==0.6.4
+    - pip install -U ray[azure]
 
 # Custom commands that will be run on worker nodes after common setup.
 worker_setup_commands: []

diff --git a/python/ray/autoscaler/azure/example-full.yaml b/python/ray/autoscaler/azure/example-full.yaml
@@ -165,7 +165,7 @@ setup_commands: []
 
 # Custom commands that will be run on the head node after common setup.
 head_setup_commands:
-    - pip install -U azure-cli-core==2.29.1 azure-identity==1.7.0 azure-mgmt-compute==23.1.0 azure-mgmt-network==19.0.0 azure-mgmt-resource==20.0.0 msrestazure==0.6.4
+    - pip install -U ray[azure]
 
 # Custom commands that will be run on worker nodes after common setup.
 worker_setup_commands: []

diff --git a/python/ray/autoscaler/azure/example-gpu-docker.yaml b/python/ray/autoscaler/azure/example-gpu-docker.yaml
@@ -118,7 +118,7 @@ setup_commands: []
 
 # Custom commands that will be run on the head node after common setup.
 head_setup_commands:
-    - pip install -U azure-cli-core==2.29.1 azure-identity==1.7.0 azure-mgmt-compute==23.1.0 azure-mgmt-network==19.0.0 azure-mgmt-resource==20.0.0 msrestazure==0.6.4
+    - pip install -U ray[azure]
 
 # Custom commands that will be run on worker nodes after common setup.
 worker_setup_commands: []

diff --git a/python/ray/autoscaler/azure/example-minimal.yaml b/python/ray/autoscaler/azure/example-minimal.yaml
@@ -19,3 +19,7 @@ auth:
     ssh_private_key: ~/.ssh/id_rsa
     # changes to this should match what is specified in file_mounts
     ssh_public_key: ~/.ssh/id_rsa.pub
+
+# Custom commands that will be run on worker nodes after common setup.
+head_setup_commands:
+  - pip install -U ray[azure]
diff --git a/python/setup.py b/python/setup.py
@@ -235,6 +235,14 @@ def get_packages(self):
         "adag": [
             "cupy-cuda12x; sys_platform != 'darwin'",
         ],
+        "azure": [
+            "azure-cli-core",
+            "azure-identity",
+            "azure-mgmt-compute",
+            "azure-mgmt-network",
+            "azure-mgmt-resource",
+            "msrestazure",
+        ],
         "client": [
             # The Ray client needs a specific range of gRPC to work:
             # Tracking issues: https://github.com/grpc/grpc/issues/33714