test_requirements.txt

tox==3.24.5
numpy==2.0
build[virtualenv]==1.2.1
flake8==4.0.1
pytest==6.2.5
pytest-cov==3.0.0
pytest-rerunfailures==10.2
pytest-timeout==2.1.0
pytest-xdist==2.4.0
coverage>=5.2, <6.2
mock==4.0.3
contextlib2==21.6.0
awslogs==0.14.0
black==24.3.0
stopit==1.1.2
# Update tox.ini to have correct version of airflow constraints file
apache-airflow==2.9.3
apache-airflow-providers-amazon==7.2.1
attrs>=23.1.0,<24
fabric==2.6.0
requests==2.32.2
sagemaker-experiments==0.1.35
Jinja2==3.1.4
pyvis==0.2.1
pandas==2.2.3
scikit-learn==1.3.0
cloudpickle==2.2.1
jsonpickle<4.0.0
PyYAML==6.0
# TODO find workaround
xgboost>=1.6.2,<=1.7.6
pillow>=10.0.1,<=11
opentelemetry-proto==1.27.0
protobuf==4.25.5
tensorboard>=2.9.0,<=2.15.2
transformers==4.46.1
sentencepiece==0.1.99
# https://github.com/triton-inference-server/server/issues/6246
tritonclient[http]<2.37.0
onnx==1.17.0
# tf2onnx==1.15.1
nbformat>=5.9,<6
accelerate>=0.24.1,<=0.27.0
schema==0.7.5
tensorflow>=2.9.0,<=2.15.1
mlflow>=2.12.2,<2.13
huggingface_hub==0.26.2
uvicorn>=0.30.1
fastapi==0.115.4
nest-asyncio
sagemaker-mlflow>=0.1.0
deepdiff>=8.0.0