From fe0865e9b7521d506964f74384f4a8cafb01750f Mon Sep 17 00:00:00 2001
From: Chen Sun <chensun@users.noreply.github.com>
Date: Wed, 3 Nov 2021 15:51:21 -0700
Subject: [PATCH] test: fix kubeflow-pipelines-tfx-python37 test (#6862)

---
 test/presubmit-tests-tfx.sh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/test/presubmit-tests-tfx.sh b/test/presubmit-tests-tfx.sh
index 3c89ac71a2..8118d659b3 100755
--- a/test/presubmit-tests-tfx.sh
+++ b/test/presubmit-tests-tfx.sh
@@ -19,6 +19,8 @@ source_root=$(pwd)
 python3 -m pip install --upgrade pip==20.2.3
 # TODO: unpin google-cloud-bigquery once TFX revert https://github.com/tensorflow/tfx/commit/f8c1dea2095197ceda60e1c4d67c4c90fc17ed44
 python3 -m pip install --upgrade google-cloud-bigquery==1.28.0
+# TODO(#6860): unpin keras version
+python3 -m pip install --upgrade keras==2.6.0
 python3 -m pip install -r "$source_root/sdk/python/requirements.txt"
 # Additional dependencies
 #pip3 install coverage==4.5.4 coveralls==1.9.2 six>=1.13.0