app_config_oom.yaml

base_image: {{ env["RAY_IMAGE_ML_NIGHTLY_GPU"] }}

env_vars: {"RAY_task_oom_retries": "50", "RAY_min_memory_free_bytes": "1000000000"}

debian_packages:
  - curl

python:
  pip_packages:
    - boto3
  conda_packages: []

post_build_cmds:
  - pip3 uninstall ray -y || true && pip3 install -U {{ env["RAY_WHEELS"] | default("ray") }}
  # TODO (Alex): We need to do this because the ray-ml image pins
  # tensorflow=2.6, which requires numpy~=1.19.2. This is ok because the test
  # doesn't actually use tensorflow, but in the long term, but we should
  # consider upgrading to tensorflow 2.7 as a long term solution.
  - pip install -U numpy>=1.20
  - {{ env["RAY_WHEELS_SANITY_CHECK"] | default("echo No Ray wheels sanity check") }}