Pin TF and Keras for 2.17 branch.

Author: arcra

.github/workflows/ci.yml

@@ -30,7 +30,11 @@ env:
   BUILDTOOLS_VERSION: '3.0.0'
   BUILDIFIER_SHA256SUM: 'e92a6793c7134c5431c58fbc34700664f101e5c9b1c1fcd93b97978e8b7f88db'
   BUILDOZER_SHA256SUM: '3d58a0b6972e4535718cdd6c12778170ea7382de7c75bc3728f5719437ffb84d'
-  TENSORFLOW_VERSION: 'tf-nightly'
+  TENSORFLOW_VERSION: 'tf-nightly==2.17.0.dev20240605'
+  # tf-nightly should install keras-nightly, but for a release we want to also pin the keras
+  # version to the closest one to TF branch cut, to minimize risk of issues with their release
+  # as well.
+  KERAS_VERSION: 'keras-nightly==3.3.3.dev2024060503'
 
 jobs:
   build:
@@ -77,6 +81,8 @@ jobs:
         run: |
           python -m pip install -U pip
           pip install "${TENSORFLOW_VERSION}"
+          # See comment where this constant is defined, above.
+          pip install "${KERAS_VERSION}"
         if: matrix.tf_version_id != 'notf'
       - name: 'Install Python dependencies'
         run: |