From 10bde261cd0bdbba5a9dc2a487ddda845710416f Mon Sep 17 00:00:00 2001 From: can Date: Tue, 16 Jul 2024 05:31:13 +0000 Subject: [PATCH] [py12] core tests Signed-off-by: can --- .buildkite/core.rayci.yml | 1 + python/requirements/ml/dl-cpu-requirements.txt | 14 +++++++------- python/requirements/ml/dl-gpu-requirements.txt | 8 ++++---- 3 files changed, 12 insertions(+), 11 deletions(-) diff --git a/.buildkite/core.rayci.yml b/.buildkite/core.rayci.yml index 77aef70d0fee8..4211129c5bfb0 100644 --- a/.buildkite/core.rayci.yml +++ b/.buildkite/core.rayci.yml @@ -24,6 +24,7 @@ steps: wanda: ci/docker/core.build.wanda.yaml matrix: - "3.10" + - "3.12" env: PYTHON: "{{matrix}}" depends_on: oss-ci-base_build-multipy diff --git a/python/requirements/ml/dl-cpu-requirements.txt b/python/requirements/ml/dl-cpu-requirements.txt index 2477e5507665e..9a281bce99174 100644 --- a/python/requirements/ml/dl-cpu-requirements.txt +++ b/python/requirements/ml/dl-cpu-requirements.txt @@ -1,13 +1,13 @@ # These requirements are used for the CI and CPU-only Docker images so we install CPU only versions of torch. # For GPU Docker images, you should install dl-gpu-requirements.txt afterwards. -tensorflow==2.15.1; sys_platform != 'darwin' or platform_machine != 'arm64' -tensorflow-macos==2.15.1; sys_platform == 'darwin' and platform_machine == 'arm64' -tensorflow-probability==0.23.0 -tensorflow-io-gcs-filesystem==0.31.0 -tensorflow-datasets -array-record==0.5.1; sys_platform != 'darwin' and platform_system != "Windows" -etils==1.5.2 +tensorflow==2.15.1; sys_platform != 'darwin' or platform_machine != 'arm64'; python_version < '3.12' +tensorflow-macos==2.15.1; sys_platform == 'darwin' and platform_machine == 'arm64'; python_version < '3.12' +tensorflow-probability==0.23.0; python_version < '3.12' +tensorflow-io-gcs-filesystem==0.31.0; python_version < '3.12' +tensorflow-datasets; python_version < '3.12' +array-record==0.5.1; sys_platform != 'darwin' and platform_system != "Windows"; python_version < '3.12' +etils==1.5.2; python_version < '3.12' # If you make changes below this line, please also make the corresponding changes to `dl-gpu-requirements.txt` # and to `install-dependencies.sh`! diff --git a/python/requirements/ml/dl-gpu-requirements.txt b/python/requirements/ml/dl-gpu-requirements.txt index ba7493f90495d..7b0a1abbc1f7f 100644 --- a/python/requirements/ml/dl-gpu-requirements.txt +++ b/python/requirements/ml/dl-gpu-requirements.txt @@ -1,9 +1,9 @@ # If you make changes below this line, please also make the corresponding changes to `dl-cpu-requirements.txt`! -tensorflow==2.15.1; sys_platform != 'darwin' or platform_machine != 'arm64' -tensorflow-macos==2.15.1; sys_platform == 'darwin' and platform_machine == 'arm64' -tensorflow-probability==0.23.0 -tensorflow-datasets +tensorflow==2.15.1; sys_platform != 'darwin' or platform_machine != 'arm64'; python_version < '3.12' +tensorflow-macos==2.15.1; sys_platform == 'darwin' and platform_machine == 'arm64'; python_version < '3.12' +tensorflow-probability==0.23.0; python_version < '3.12' +tensorflow-datasets; python_version < '3.12' --extra-index-url https://download.pytorch.org/whl/cu121 # for GPU versions of torch, torchvision --find-links https://data.pyg.org/whl/torch-2.3.0+cu121.html # for GPU versions of torch-scatter, torch-sparse, torch-cluster, torch-spline-conv