[py12] core tests

Signed-off-by: can <can@anyscale.com>

.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

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`!

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