tfmot package requirements

PiperOrigin-RevId: 246387230

Author: alanchiao

requirements.txt

@@ -1 +1,3 @@
-numpy
+absl-py~=0.7
+numpy~=1.14
+six~=1.10

setup.py

@@ -33,7 +33,10 @@
 # one of the two) to one where a user installs the tf package and then
 # also installs the gpu package if they need gpu support. The latter allows
 # us (and our dependents) to maintain a single package instead of two.
-REQUIRED_PACKAGES = []
+REQUIRED_PACKAGES = [
+    'numpy~=1.14',
+    'six~=1.10',
+]
 
 if '--release' in sys.argv:
   release = True