From a9ea9ba4a00b436d7dcd8ea2060d03f5c2bb3bb0 Mon Sep 17 00:00:00 2001
From: Michael Reneer <michaelreneer@google.com>
Date: Tue, 10 Oct 2023 10:33:37 -0700
Subject: [PATCH] Remove unneeded `tensorflow-privacy` dependencies.

Please note:
1. I have manually tested the TFP release for python 3.10.
2. TFP does not define the set of python version that it is compatible with.

This change should be followed up with the following changes:
* Define the set of python version that TFP is compatible with.
* Setup a Python package building test.
* Release TFP so that projects depending on TFP can support Python 3.11

PiperOrigin-RevId: 572297305
---
 README.md                 | 8 --------
 requirements.txt          | 1 -
 setup.py                  | 9 +++------
 tools/test_pip_package.sh | 2 +-
 4 files changed, 4 insertions(+), 16 deletions(-)

diff --git a/README.md b/README.md
index f4116425..73f2edb0 100644
--- a/README.md
+++ b/README.md
@@ -31,14 +31,6 @@ better performance, it is also recommended to install TensorFlow with GPU
 support (detailed instructions on how to do this are available in the TensorFlow
 installation documentation).
 
-In addition to TensorFlow and its dependencies, other prerequisites are:
-
-*   `scipy` >= 0.17
-
-*   `mpmath` (for testing)
-
-*   `tensorflow_datasets` (for the RNN tutorial `lm_dpsgd_tutorial.py` only)
-
 ### Installing TensorFlow Privacy
 
 If you only want to use TensorFlow Privacy as a library, you can simply execute
diff --git a/requirements.txt b/requirements.txt
index aa27f59d..27f0fb8b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -26,7 +26,6 @@ dm-tree==0.1.8
 dp-accounting==0.4.3
 immutabledict~=2.2
 matplotlib~=3.3
-mpmath~=1.2
 numpy~=1.21
 packaging~=22.0
 pandas~=1.4
diff --git a/setup.py b/setup.py
index cd623a1b..45a20ef9 100644
--- a/setup.py
+++ b/setup.py
@@ -13,15 +13,14 @@
 # limitations under the License.
 """TensorFlow Privacy library setup file for pip."""
 
-from setuptools import find_packages
-from setuptools import setup
+import setuptools
 
 with open('tensorflow_privacy/version.py') as file:
   globals_dict = {}
   exec(file.read(), globals_dict)  # pylint: disable=exec-used
   VERSION = globals_dict['__version__']
 
-setup(
+setuptools.setup(
     name='tensorflow_privacy',
     version=VERSION,
     url='https://github.com/tensorflow/privacy',
@@ -36,15 +35,13 @@
         'numpy~=1.21',
         'packaging~=22.0',
         'pandas~=1.4',
-        'parameterized~=0.8',
         'scikit-learn>=1.0,==1.*',
         'scipy~=1.9',
         'statsmodels~=0.13',
-        'tensorflow-datasets~=4.5',
         'tensorflow-estimator~=2.4',
         'tensorflow-probability~=0.22.0',
         'tensorflow~=2.4',
         'tf-models-official~=2.13',
     ],
-    packages=find_packages(),
+    packages=setuptools.find_packages(),
 )
diff --git a/tools/test_pip_package.sh b/tools/test_pip_package.sh
index 8dc075fd..afdc4b71 100755
--- a/tools/test_pip_package.sh
+++ b/tools/test_pip_package.sh
@@ -28,7 +28,7 @@ main() {
   pip install --upgrade pip
   pip --version
 
-  # Publish the pip package.
+  # Test the pip package.
   package="$(ls "dist/"*".whl" | head -n1)"
   pip install --upgrade "${package}"
   pip freeze