From 0687f677f6979d36e82860fb84c5f3d5006b8231 Mon Sep 17 00:00:00 2001 From: Erick Benitez-Ramos Date: Wed, 13 Nov 2024 07:17:04 +0000 Subject: [PATCH 1/3] fix: pin testing deps --- requirements/extras/test_requirements.txt | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/requirements/extras/test_requirements.txt b/requirements/extras/test_requirements.txt index a2c0fbfc65..7f32b53bc3 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.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.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 From c039a11a5d31983cdcd93e67584a26823bd4b33f Mon Sep 17 00:00:00 2001 From: Erick Benitez-Ramos Date: Wed, 13 Nov 2024 07:24:38 +0000 Subject: [PATCH 2/3] adjust deps --- requirements/extras/test_requirements.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements/extras/test_requirements.txt b/requirements/extras/test_requirements.txt index 7f32b53bc3..d55dbe764b 100644 --- a/requirements/extras/test_requirements.txt +++ b/requirements/extras/test_requirements.txt @@ -32,7 +32,7 @@ xgboost>=1.6.2,<=1.7.6 pillow>=10.0.1,<=11 opentelemetry-proto==1.27.0 protobuf==4.25.5 -tensorboard==2.15.2 +tensorboard>=2.9.0,<=2.15.2 transformers==4.46.1 sentencepiece==0.1.99 # https://github.com/triton-inference-server/server/issues/6246 @@ -42,7 +42,7 @@ onnx==1.17.0 nbformat>=5.9,<6 accelerate>=0.24.1,<=0.27.0 schema==0.7.5 -tensorflow==2.15.1 +tensorflow=>2.9.0,<=2.15.1 mlflow>=2.12.2,<2.13 huggingface_hub==0.26.2 uvicorn>=0.30.1 From a2f854592abdb4c5ff707541f3b510e036d785ab Mon Sep 17 00:00:00 2001 From: Erick Benitez-Ramos Date: Wed, 13 Nov 2024 07:37:51 +0000 Subject: [PATCH 3/3] fix pin --- requirements/extras/test_requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements/extras/test_requirements.txt b/requirements/extras/test_requirements.txt index d55dbe764b..1592576a47 100644 --- a/requirements/extras/test_requirements.txt +++ b/requirements/extras/test_requirements.txt @@ -42,7 +42,7 @@ onnx==1.17.0 nbformat>=5.9,<6 accelerate>=0.24.1,<=0.27.0 schema==0.7.5 -tensorflow=>2.9.0,<=2.15.1 +tensorflow>=2.9.0,<=2.15.1 mlflow>=2.12.2,<2.13 huggingface_hub==0.26.2 uvicorn>=0.30.1