diff --git a/requirements/extras/test_requirements.txt b/requirements/extras/test_requirements.txt index a2c0fbfc65..1592576a47 100644 --- a/requirements/extras/test_requirements.txt +++ b/requirements/extras/test_requirements.txt @@ -22,26 +22,30 @@ requests==2.32.2 sagemaker-experiments==0.1.35 Jinja2==3.1.4 pyvis==0.2.1 -pandas>=1.3.5,<1.5 +pandas==1.4.4 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 -transformers>=4.36.0 +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.15.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.1,<=2.16 +tensorflow>=2.9.0,<=2.15.1 mlflow>=2.12.2,<2.13 -huggingface_hub>=0.23.4 +huggingface_hub==0.26.2 uvicorn>=0.30.1 -fastapi>=0.111.0 +fastapi==0.115.4 nest-asyncio sagemaker-mlflow>=0.1.0