From d08a0cbd19d709c426fcf346c96cf3ba327ef7f4 Mon Sep 17 00:00:00 2001
From: Ebrahim Songhori <esonghori@google.com>
Date: Wed, 15 Nov 2023 13:33:41 -0800
Subject: [PATCH] Adding tf-keras to the requirements.

PiperOrigin-RevId: 582788811
Change-Id: Iabbc2b394d42c8ffe0a79db155e87cc1fee31f6c
---
 tox.ini | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tox.ini b/tox.ini
index 7a0e51a..1c392f6 100644
--- a/tox.ini
+++ b/tox.ini
@@ -16,6 +16,7 @@ deps = pytest
        numpy>=1.15.4
        torch==1.13.1
        shapely>=1.7.0
+       tf_keras>=2.15.0
 
 [stable]
 deps = pytest
@@ -33,6 +34,7 @@ deps = pytest
        numpy>=1.15.4
        torch==1.13.1
        shapely>=1.7.0
+       tf_keras>=2.15.0
 
 [testenv:py39-stable]
 skip_install = True