Left from previous fix.

PiperOrigin-RevId: 583115192 Change-Id: I2b9609ac5ef7881cc442bbfd239ace63f6894453
tensorflow · Nov 16, 2023 · 0c336d6 · 0c336d6
1 parent 2a18743
commit 0c336d6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tools/docker/ubuntu_tf_agents b/tools/docker/ubuntu_tf_agents
@@ -87,7 +87,8 @@ RUN $python_version -m pip install wheel \
       protobuf==4.21.12 \
       google-api-core \
       google-cloud-bigquery \
-      psutil
+      psutil \
+      tf-keras
 
 ######################################################
 # Installs Jupyter notebook.