From c16a217c3334373f4071e0234efb4d871c433166 Mon Sep 17 00:00:00 2001
From: Ping Yu <4018+pyu10055@users.noreply.github.com>
Date: Wed, 3 Nov 2021 14:09:12 -0700
Subject: [PATCH] lock keras version (#5809)

BUG
---
 tfjs-converter/python/BUILD            | 1 +
 tfjs-converter/python/requirements.txt | 1 +
 2 files changed, 2 insertions(+)

diff --git a/tfjs-converter/python/BUILD b/tfjs-converter/python/BUILD
index 772609676d6..5b2e7f12ccf 100644
--- a/tfjs-converter/python/BUILD
+++ b/tfjs-converter/python/BUILD
@@ -48,6 +48,7 @@ py_wheel(
     license = "Apache 2.0",
     python_tag = "py3",
     requires = [
+        "keras<=2.6.0",
         "tensorflow-estimator<=2.6.0",
         "tensorflow>=2.1.0,<3",
         "six>=1.12.0,<2",
diff --git a/tfjs-converter/python/requirements.txt b/tfjs-converter/python/requirements.txt
index 776f041a9de..3519ab7b92d 100644
--- a/tfjs-converter/python/requirements.txt
+++ b/tfjs-converter/python/requirements.txt
@@ -1,3 +1,4 @@
+keras<=2.6.0
 tensorflow-estimator<=2.6.0
 tensorflow>=2.1.0,<3
 protobuf==3.17.3; python_version < "3"