# Unique identifier for the head node and workers of this cluster.
cluster_name: nightly-cpu-minimal-2
max_workers: 6
idle_timeout_minutes: 5

# Cloud-provider specific configuration.
provider:
  type: azure
    # https://azure.microsoft.com/en-us/global-infrastructure/locations
  location: westus2
  resource_group: ray-zones
  cache_stopped_nodes: False
  # Provider-level availability zone configuration (comma-separated)
  # This will be used as default for all node types unless overridden
  availability_zone: "1,2,3"

auth:
  ssh_user: ubuntu

available_node_types:
  ray.head.default:
    resources: {"CPU": 2}
    node_config:
      azure_arm_parameters:
        vmSize: Standard_D2s_v3
        imagePublisher: microsoft-dsvm
        imageOffer: ubuntu-2204
        imageSku: 2204-gen2
        imageVersion: latest
        # Head node: explicitly disable availability zones
        availability_zone: "none"
  ray.worker.default:
    min_workers: 0
    max_workers: 2
    resources: {"CPU": 2}
    node_config:
      azure_arm_parameters:
        vmSize: Standard_D2s_v3
        imagePublisher: microsoft-dsvm
        imageOffer: ubuntu-2204
        imageSku: 2204-gen2
        imageVersion: latest
        # Workers will use provider specified availability zones
  ray.worker.specific_zone:
    min_workers: 0
    max_workers: 2
    resources: {"CPU": 2}
    node_config:
      azure_arm_parameters:
        vmSize: Standard_D2s_v3
        imagePublisher: microsoft-dsvm
        imageOffer: ubuntu-2204
        imageSku: 2204-gen2
        imageVersion: latest
        # Workers will use availability zone 2 only (overrides provider setting)
        availability_zone: "2"

# Note: The Ubuntu 20.04 dsvm image has a few venvs already configured but
# they all contain python modules that are not compatible with Ray at the moment.
setup_commands:
    - (which conda && echo 'eval "$(conda shell.bash hook)"' >> ~/.bashrc) || true
    - conda tos accept
    - conda create -n ray-env python=3.10 -y
    - conda activate ray-env && echo 'conda activate ray-env' >> ~/.bashrc
    - which ray || pip install -U "ray[default] @ https://s3-us-west-2.amazonaws.com/ray-wheels/latest/ray-3.0.0.dev0-cp310-cp310-manylinux2014_x86_64.whl"

file_mounts_sync_continuously: False

file_mounts: {
}