Removing tf-keras dependency as it is installed as the dependency o…

…f tensorflow. Also, its version should match the tensorflow's version. PiperOrigin-RevId: 587882239 Change-Id: Ib144e6ded4c27d34ae548c2080502f60d2f7a1ef
tensorflow · Dec 5, 2023 · 06705c6 · 06705c6
1 parent c67f483
commit 06705c6
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -189,7 +189,6 @@ def get_required_packages():
       # installed or if gym[*] (where * is an environment which lists pygame as
       # a dependency).
       'pygame == 2.1.3',
-      'tf-keras >= 2.15.0',
   ]
   add_additional_packages(required_packages)
   return required_packages

diff --git a/tools/docker/ubuntu_tf_agents b/tools/docker/ubuntu_tf_agents
@@ -87,8 +87,7 @@ RUN $python_version -m pip install wheel \
       protobuf==4.21.12 \
       google-api-core \
       google-cloud-bigquery \
-      psutil \
-      tf-keras
+      psutil
 
 ######################################################
 # Installs Jupyter notebook.