FROM python:3.9

# -------------------------------------------------------------------
# Install Ray and essential packages.
# For more information on Ray installation, see:
#   https://docs.ray.io/en/latest/ray-overview/installation.html
# Install the latest Dask versions that are compatible with
#   Ray nightly. For more information, see:
#   https://docs.ray.io/en/latest/data/dask-on-ray.html
# -------------------------------------------------------------------

RUN pip install --no-cache-dir \
"ray==2.8.0" \
"ray[default]==2.8.0" \
"dask==2022.10.1" \
torch \
"pydantic<2" \
six \
"tqdm<2" \
"pandas<2" \
scikit-learn \
matplotlib \
optuna \
seaborn \
tabulate \
jupyterlab \
autopep8