hunk.code_window: # Needed for Ray Client error message serialization/deserialization.
   tblib
   # In TF >v2, GPU support is included in the base package.
  
hunk.labels: keep keep keep add keep
hunk.after_edit: # If you make changes to anything below this line, please also change `requirements_dl.txt`!
   # TODO(amogkam): Remove after https://github.com/tensorflow/tensorflow/issues/52922 is fixed.
   keras==2.6.0
  
hunk.file_path: python/requirements_ml_docker.txt
hunk.type: add
file: cloud_id: {{env["ANYSCALE_CLOUD_ID"]}}
  region: us-west-2
  
  max_workers: 999
  
  head_node_type:
      name: head_node
      instance_type: r5dn.16xlarge # Network optimized.
      resources:
        cpu: 0
        custom_resources:
          node: 1
          small: 1
  
  worker_node_types:
      - name: worker_node_m
        instance_type: m5.16xlarge
        min_workers: 32
        max_workers: 32
        use_spot: false
        resources:
          custom_resources:
            node: 1
      - name: worker_node_r
        instance_type: r5.16xlarge
        min_workers: 32
        max_workers: 32
        use_spot: false
        resources:
          custom_resources:
            node: 1
  