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

PiperOrigin-RevId: 625814369

Author: tensorflower-gardener

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: