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

tensorflower-gardener · tensorflower-gardener · commit 7557794b37d7 · 2024-04-17T15:34:14.000-07:00
PiperOrigin-RevId: 625826714
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:
