diff --git a/containers/conda_env/aarch64_test.yml b/containers/conda_env/aarch64_test.yml index 14305eb..d7dd136 100644 --- a/containers/conda_env/aarch64_test.yml +++ b/containers/conda_env/aarch64_test.yml @@ -26,7 +26,7 @@ dependencies: - awscli - numba - llvmlite -- loky +- loky>=3.5.1 - pyarrow - pyspark>=3.4.0 - cloudpickle diff --git a/containers/conda_env/linux_cpu_test.yml b/containers/conda_env/linux_cpu_test.yml index 1ec2a54..1e8310c 100644 --- a/containers/conda_env/linux_cpu_test.yml +++ b/containers/conda_env/linux_cpu_test.yml @@ -33,7 +33,7 @@ dependencies: - boto3 - awscli - py-ubjson -- loky +- loky>=3.5.1 - pyarrow - protobuf - cloudpickle diff --git a/containers/dockerfile/Dockerfile.gpu b/containers/dockerfile/Dockerfile.gpu index 7794703..b48910f 100644 --- a/containers/dockerfile/Dockerfile.gpu +++ b/containers/dockerfile/Dockerfile.gpu @@ -36,7 +36,7 @@ RUN \ distributed \ "dask-cuda=$RAPIDS_VERSION.*" "dask-cudf=$RAPIDS_VERSION.*" cupy \ numpy pytest pytest-timeout scipy scikit-learn pandas matplotlib wheel \ - python-kubernetes urllib3 graphviz hypothesis loky \ + python-kubernetes urllib3 graphviz hypothesis "loky>=3.5.1" \ "pyspark>=3.4.0" cloudpickle cuda-python && \ mamba clean --all --yes