diff --git a/requirements/pytorch/test.txt b/requirements/pytorch/test.txt index 485e6e6fd0393..ec1d68ccb8459 100644 --- a/requirements/pytorch/test.txt +++ b/requirements/pytorch/test.txt @@ -8,7 +8,7 @@ pre-commit==2.20.0 # needed in tests cloudpickle>=1.3, <2.3.0 -scikit-learn>0.22.1, <1.1.3 +scikit-learn>0.22.1, <1.2.1 onnxruntime<1.14.0 psutil<5.9.4 # for `DeviceStatsMonitor` pandas>1.0, <1.5.2 # needed in benchmarks