- Fixing dependencies in setup.py and requirements.txt.

PiperOrigin-RevId: 227742524

README.md

@@ -20,8 +20,9 @@ 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).
 
-Installing TensorFlow will take care of all other dependencies like `numpy` and
-`scipy`.
+In addition to TensorFlow and its dependencies, other prerequisites are:
+  * `scipy` >= 0.17
+  * `mpmath` (for testing)
 
 ### Installing TensorFlow Privacy
 

privacy/analysis/rdp_accountant_test.py

@@ -27,6 +27,7 @@ import numpy as np
 
 from privacy.analysis import rdp_accountant
 
+
 class TestGaussianMoments(parameterized.TestCase):
   #################################
   # HELPER FUNCTIONS:             #

requirements.txt

@@ -1 +1,3 @@
 tensorflow
+mpmath
+scipy

setup.py

@@ -18,10 +18,14 @@ setup(name='privacy',
       version='0.0.1',
       url='https://github.com/tensorflow/privacy',
       license='Apache-2.0',
+      install_requires=[
+          'scipy>=0.17',
+          'mpmath',  # used in tests only
+      ],
       # Explicit dependence on TensorFlow is not supported.
       # See https://github.com/tensorflow/tensorflow/issues/7166
       extras_require={
-          "tf": ["tensorflow>=1.0.0"],
-          "tf_gpu": ["tensorflow-gpu>=1.0.0"],
+          'tf': ['tensorflow>=1.0.0'],
+          'tf_gpu': ['tensorflow-gpu>=1.0.0'],
       },
       packages=find_packages())