Install tf-keras to avoid stable failure until the tf-agent stable is…

… updated.

PiperOrigin-RevId: 593009210
Change-Id: Ibc7c212f64937edbf254873da16e239b67bd2c58

README.md

@@ -161,6 +161,7 @@ These commands install TF-Agents and the placement cost binary.
 ```shell
 # Installs TF-Agents with stable versions of Reverb and TensorFlow 2.x.
 $  pip install $TF_AGENTS_PIP_VERSION
+$  pip install tf-keras
 # Using keras-2
 $ export TF_USE_LEGACY_KERAS=1
 # Copies the placement cost binary to /usr/local/bin and makes it executable.
@@ -250,6 +251,7 @@ These commands install TF-Agents and the placement cost binary.
 ```shell
 # Installs TF-Agents with stable versions of Reverb and TensorFlow 2.x.
 $  pip install tf-agents-nightly[reverb]
+$  pip install tf-keras
 # Using keras-2
 $ export TF_USE_LEGACY_KERAS=1
 # Copies the placement cost binary to /usr/local/bin and makes it executable.

tools/docker/ubuntu_circuit_training

@@ -84,7 +84,8 @@ RUN $python_version -mpip install pyunpack>=0.1.2 \
         scipy>=1.1.0 \
         numpy>=1.15.4 \
         torch==1.13.1 \
-        shapely>=1.7.0
+        shapely>=1.7.0 \
+        tf-keras>=2.15.0
 
 # Installs TF-Agents and Tensorflow along with tox and pytest to help verify the
 # container via unit tests.

tox.ini

@@ -5,6 +5,7 @@ deps = pytest
        tf-agents-nightly[reverb]
        sortedcontainers
        timeout-decorator>=0.5.0
+       tf-keras>=2.15.0
        # Neded by DREAMPlace
        pyunpack>=0.1.2
        patool>=1.12
@@ -22,6 +23,7 @@ deps = pytest
        tf-agents[reverb]
        sortedcontainers
        timeout-decorator>=0.5.0
+       tf-keras>=2.15.0
        # Neded by DREAMPlace
        pyunpack>=0.1.2
        patool>=1.12