Add required packages to setup.py to match requirements.txt

PiperOrigin-RevId: 625814369
tensorflow · Apr 17, 2024 · eb62857 · eb62857
1 parent 73851a0
commit eb62857
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -36,8 +36,11 @@
     'absl-py~=1.2',
     'numpy~=1.23',
     'six~=1.14',
+    'scipy',
     'enum34~=1.1;python_version<"3.4"',
     'dm-tree~=0.1.1',
+    'mock',
+    'tf-keras>=2.14.1'
 ]
 
 if '--release' in sys.argv: