diff --git a/.ci/docker/requirements-ci.txt b/.ci/docker/requirements-ci.txt index a79fb413902cc..b3523a891f53e 100644 --- a/.ci/docker/requirements-ci.txt +++ b/.ci/docker/requirements-ci.txt @@ -113,11 +113,7 @@ ninja==1.11.1.3 #Pinned versions: 1.11.1.3 #test that import: run_test.py, test_cpp_extensions_aot.py,test_determination.py -numba==0.49.0 ; python_version < "3.9" -numba==0.55.2 ; python_version == "3.9" -numba==0.55.2 ; python_version == "3.10" -numba==0.60.0 ; python_version == "3.12" -#Description: Just-In-Time Compiler for Numerical Functions +numba==0.61.2 #Pinned versions: 0.54.1, 0.49.0, <=0.49.1 #test that import: test_numba_integration.py #For numba issue see https://github.com/pytorch/pytorch/issues/51511 @@ -134,13 +130,9 @@ numba==0.60.0 ; python_version == "3.12" #test_nn.py, test_namedtensor.py, test_linalg.py, test_jit_cuda_fuser.py, #test_jit.py, test_indexing.py, test_datapipe.py, test_dataloader.py, #test_binary_ufuncs.py -numpy==1.22.4; python_version == "3.9" or python_version == "3.10" -numpy==1.26.2; python_version == "3.11" or python_version == "3.12" -numpy==2.1.2; python_version >= "3.13" - -pandas==2.0.3; python_version < "3.13" -pandas==2.2.3; python_version >= "3.13" +numpy==2.1.2 +pandas==2.2.3 #onnxruntime #Description: scoring engine for Open Neural Network Exchange (ONNX) models #Pinned versions: 1.9.0 diff --git a/requirements.txt b/requirements.txt index 39a00bdbd5b5e..19022d173aa74 100644 --- a/requirements.txt +++ b/requirements.txt @@ -9,9 +9,7 @@ jinja2 lintrunner ; platform_machine != "s390x" networkx ninja -numpy==1.22.4; python_version == "3.9" or python_version == "3.10" -numpy==1.26.2; python_version == "3.11" or python_version == "3.12" -numpy==2.1.2; python_version >= "3.13" +numpy==2.1.2 optree>=0.13.0 packaging psutil