diff --git a/dockerfiles/base/requirements.txt b/dockerfiles/base/requirements.txt index d1eb0f426ce..1a5e6992b92 100644 --- a/dockerfiles/base/requirements.txt +++ b/dockerfiles/base/requirements.txt @@ -3,7 +3,7 @@ urllib3>=1.25.4, <1.27 blosc~=1.7 cloudpickle~=1.5 dask-kubernetes~=0.11.0 -dask-ml~=1.4 +dask-ml~=1.4,<1.9.0 dask[complete]~=2.12 # see main requirements.txt for reasoning the resrictions distributed>=2.23, <3 diff --git a/dockerfiles/models-gpu/requirements.txt b/dockerfiles/models-gpu/requirements.txt index c60488a2fbd..bad2fa29927 100644 --- a/dockerfiles/models-gpu/requirements.txt +++ b/dockerfiles/models-gpu/requirements.txt @@ -3,7 +3,7 @@ urllib3>=1.25.4, <1.27 blosc~=1.7 cloudpickle~=1.5 dask-kubernetes~=0.11.0 -dask-ml~=1.4 +dask-ml~=1.4,<1.9.0 dask[complete]~=2.12 # see main requirements.txt for reasoning the resrictions distributed>=2.23, <3 diff --git a/dockerfiles/models/requirements.txt b/dockerfiles/models/requirements.txt index eefaa92282e..00275ff7b6c 100644 --- a/dockerfiles/models/requirements.txt +++ b/dockerfiles/models/requirements.txt @@ -4,7 +4,7 @@ blosc~=1.7 bokeh~=2.0 cloudpickle~=1.5 dask-kubernetes~=0.11.0 -dask-ml~=1.4 +dask-ml~=1.4,<1.9.0 dask[complete]~=2.12 # see main requirements.txt for reasoning the resrictions distributed>=2.23, <3