[breaking] Bump Python requirement to 3.10. (#10434)
- Bump the Python requirement. - Fix type hints. - Use loky to avoid deadlock. - Workaround cupy-numpy compatibility issue on Windows caused by the `safe` casting rule. - Simplify the repartitioning logic to avoid dask errors.
This commit is contained in:
@@ -30,7 +30,8 @@ RUN \
|
||||
"nccl>=${NCCL_SHORT_VER}" \
|
||||
dask \
|
||||
"dask-cuda=$RAPIDS_VERSION_ARG.*" "dask-cudf=$RAPIDS_VERSION_ARG.*" cupy \
|
||||
numpy pytest pytest-timeout scipy scikit-learn pandas matplotlib wheel python-kubernetes urllib3 graphviz hypothesis \
|
||||
numpy pytest pytest-timeout scipy scikit-learn pandas matplotlib wheel \
|
||||
python-kubernetes urllib3 graphviz hypothesis loky \
|
||||
"pyspark>=3.4.0" cloudpickle cuda-python && \
|
||||
mamba clean --all --yes && \
|
||||
conda run --no-capture-output -n gpu_test pip install buildkite-test-collector
|
||||
|
||||
Reference in New Issue
Block a user