Adding tf-keras to the requirements.

PiperOrigin-RevId: 582788811 Change-Id: Iabbc2b394d42c8ffe0a79db155e87cc1fee31f6c
google-research · Nov 15, 2023 · d08a0cb · d08a0cb
1 parent caae5b8
commit d08a0cb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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