base_image: "anyscale/ray:nightly-py37-cu102"

python:
  pip_packages:
    - pyarrow<7.0.0
  conda_packages: []

post_build_cmds:
    - pip uninstall -y ray && pip3 install -U {{ env["RAY_WHEELS"] | default("ray") }}
    - pip3 install llvmlite --ignore-installed
    - pip3 install -U torch==1.6 dask dask_ml mlflow tensorboard torchvision pickle5 s3fs
    - {{ env["RAY_WHEELS_SANITY_CHECK"] | default("echo No Ray wheels sanity check") }}