Merge pull request tensorflow#69726 from tensorflow/r2.16-264dce9fb38

nitins17 · web-flow · commit dd18d134cd16 · 2024-06-13T12:11:39.000-07:00
r2.16 cherry-pick: 264dce9 "Remove CMake from requirements now that dm-tree has 3.12 wheels."
diff --git a/ci/official/containers/linux_arm64/devel.packages.txt b/ci/official/containers/linux_arm64/devel.packages.txt
@@ -3,8 +3,6 @@ autoconf
 automake
 build-essential
 ca-certificates
-# TODO(b/308399490) Remove CMake once dm-tree (Keras dependency) has 3.12 wheels
-cmake
 llvm-17
 clang-17
 clang-format-12
diff --git a/ci/official/utilities/setup_macos.sh b/ci/official/utilities/setup_macos.sh
@@ -81,13 +81,6 @@ if [[ "${TFCI_MACOS_PYENV_INSTALL_ENABLE}" == 1 ]]; then
   python --version
 fi
 
-if [[ "$TFCI_PYTHON_VERSION" == "3.12" ]]; then
-  # dm-tree (Keras v3 dependency) doesn't have pre-built wheels for 3.12 yet.
-  # Having CMake allows building them.
-  # Once the wheels are added, this should be removed - b/308399490.
-  brew install cmake
-fi
-
 # TFCI Mac VM images do not have twine installed by default so we need to
 # install it manually. We use Twine in nightly builds to upload Python packages
 # to PyPI.
diff --git a/tensorflow/tools/tf_sig_build_dockerfiles/devel.packages.txt b/tensorflow/tools/tf_sig_build_dockerfiles/devel.packages.txt
@@ -36,8 +36,6 @@ autoconf
 automake
 build-essential
 ca-certificates
-# TODO(b/308399490) Remove CMake once dm-tree (Keras dependency) has 3.12 wheels
-cmake
 llvm-17
 clang-17
 clang-tidy-17
