## Requirements for running tests
# These should all be pinned to versions that work

# General test requirements
async-exit-stack==1.0.1
async-generator==1.10
asyncmock==0.4.2
azure-cli-core==2.40.0
azure-identity==1.10.0
azure-mgmt-compute==23.1.0
azure-mgmt-network==19.0.0
azure-mgmt-resource==20.0.0
msrestazure==0.6.4
beautifulsoup4==4.11.1
boto3==1.24.59
# Todo: investigate if we can get rid of this and exchange for ray.cloudpickle
cloudpickle==2.2.0
# Keep in sync with `ci/build/upload_build_info.sh`
cryptography==38.0.1
cython==0.29.32
dataclasses; python_version < '3.7'
# 0.85.0 of fastapi is breaking the ci because
# requires newer starlette -> newer version typing-extensions
# conflicting with tensorflow 2.6
fastapi==0.75.0
feather-format==0.4.1
# Keep compatible with Werkzeug
flask==2.1.3
google-api-python-client==2.65.0
google-cloud-storage==2.5.0
gradio==3.11; platform_system != "Windows"
jsonpatch==1.32
kubernetes==24.2.0
llvmlite==0.39.1; python_version < '3.11'
lxml==4.9.1
moto[s3,server]==4.0.7
mypy==0.982
networkx==2.6.3
numba==0.56.4
openpyxl==3.0.10
opentelemetry-api==1.1.0
opentelemetry-sdk==1.1.0
opentelemetry-exporter-otlp==1.1.0
opentelemetry-exporter-opencensus==0.20b0
pexpect==4.8.0
Pillow==9.2.0; platform_system != "Windows"
proxy.py==2.4.3
pyarrow==6.0.1; python_version < '3.11'
pyarrow==10.0.1; python_version >= '3.11'
pydantic==1.9.2; python_version < '3.11'
pydantic==1.10.2; python_version >= '3.11'
# Keep in sync with `ci/build/upload_build_info.sh`
PyOpenSSL==22.1.0
pygame==2.1.2; python_version < '3.11'
Pygments==2.13.0
pymongo==4.3.2
# TODO: Upgrade to pyspark 3.4.0 once raydp supports it
# https://ml-team-public-read.s3.us-west-2.amazonaws.com/spark-pkgs/pyspark-3.4.0.dev0-0cb0fa313979e1b82ddd711a05d8c4e78cf6c9f5.tar.gz
pyspark==3.3.1
pytest==7.0.1
pytest-asyncio==0.16.0
pytest-rerunfailures==10.2
pytest-sugar==0.9.5
pytest-lazy-fixture==0.6.3
pytest-timeout==2.1.0
pytest-virtualenv==1.7.0
redis==4.4.2
scikit-learn==1.0.2; python_version < '3.11'
smart_open[s3]==6.2.0
tqdm==4.64.1
testfixtures==7.0.0
werkzeug==2.1.2
xlrd==2.0.1
memray; platform_system != "Windows" and sys_platform != "darwin"

# For doc tests
myst-parser==0.15.2
myst-nb==0.13.1
sphinx==4.3.2
jupytext==1.13.6
jinja2==3.0.3
pytest-docker-tools==3.1.3
pytest-forked==1.4.0

# For dataset tests
polars==0.14.21

# ale-py requires 4.10.0. It's also compatible with flake8.
# It's not compatible with mosaicml though. Try to remove once
# we are at Python 3.8
importlib-metadata==4.10.0

# Some packages have downstream dependencies that we have to specify here to resolve conflicts.
# Feel free to add (or remove!) packages here liberally.
tensorboardX==2.4.1
starlette==0.17.1
h11==0.12.0
markdown-it-py==1.1.0
attrs==21.4.0
pytz==2022.7.1
# Compatibility with spacy 3.5 (model en_core_web_sm)
typing-extensions==4.5.0
networkx==2.6.3; python_version <= '3.7'
# Aim requires segment-analytics-python, which requires backoff~=2.10,
# which conflicts with the opentelemetry-api 1.1.0.
segment-analytics-python==2.2.0
httpcore==0.15.0
backoff==1.10
sympy==1.10.1; python_version <= '3.7'

# For test_basic.py::test_omp_threads_set
threadpoolctl==3.1.0
numexpr==2.8.4