cluster_name: ray-tune-scalability-tests-1x32_hd

max_workers: 0
upscaling_speed: 1

idle_timeout_minutes: 0

docker:
    image: anyscale/ray:nightly
    container_name: ray_container
    pull_before_run: true

provider:
    type: aws
    region: us-west-2
    availability_zone: us-west-2a
    cache_stopped_nodes: false

available_node_types:
    cpu_32_hd_ondemand:
        node_config:
            InstanceType: m5.8xlarge

            BlockDeviceMappings:
                - DeviceName: /dev/sda1
                  Ebs:
                      VolumeSize: 202

            TagSpecifications:
                - ResourceType: "instance"
                  Tags:
                    - Key: anyscale-user
                      Value: '{{env["ANYSCALE_USER"]}}'
                    - Key: anyscale-expiration
                      Value: '{{env["ANSYCALE_EXPIRATION"]}}'

        resources: {"CPU": 32}  # 128 GB memory
        min_workers: 0
        max_workers: 0

auth:
    ssh_user: ubuntu

file_mounts: {
    "~/aws/credentials": "./aws_secrets.txt"
}

head_node_type: cpu_32_hd_ondemand
worker_default_node_type: cpu_32_hd_ondemand

setup_commands:
    - pip install -U {{env["RAY_WHEEL"]}}