From 06705c6cb1d724903ad759d7662adea204666835 Mon Sep 17 00:00:00 2001
From: Ebrahim Songhori <esonghori@google.com>
Date: Mon, 4 Dec 2023 16:53:28 -0800
Subject: [PATCH] Removing `tf-keras` dependency as it is installed as the
 dependency of tensorflow. Also, its version should match the tensorflow's
 version.

PiperOrigin-RevId: 587882239
Change-Id: Ib144e6ded4c27d34ae548c2080502f60d2f7a1ef
---
 setup.py                      | 1 -
 tools/docker/ubuntu_tf_agents | 3 +--
 2 files changed, 1 insertion(+), 3 deletions(-)

diff --git a/setup.py b/setup.py
index e8f498038..8babcf3f7 100644
--- 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
index fb443a593..8071ce38f 100644
--- 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.