## 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.23.10
# 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
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: Replace this with pyspark==3.4 once it is released.
https://ml-team-public-read.s3.us-west-2.amazonaws.com/pyspark-3.4.0.dev0.tar.gz
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
jupytext==1.13.6
pytest-docker-tools==3.1.3
pytest-forked==1.4.0

# For dataset tests
polars==0.14.21

# 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
importlib-metadata==4.13.0