From abb33ee76f9f9833c8cb89888628c4327fe4b88b Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Wed, 17 Apr 2024 14:51:41 -0700
Subject: [PATCH] Add required packages to setup.py to match requirements.txt

PiperOrigin-RevId: 625814369
---
 setup.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/setup.py b/setup.py
index 9f26fccd4..51cf66968 100644
--- 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: