Update to allow bazel on tensorflow_privacy to work out of the box.

schien1729 · npapernot · commit 0af76c7b3d04 · 2018-12-18T15:41:26.000-08:00
PiperOrigin-RevId: 225605386
diff --git a/privacy/optimizers/BUILD b/privacy/optimizers/BUILD
diff --git a/privacy/optimizers/dp_adam.py b/privacy/optimizers/dp_adam.py
@@ -20,7 +20,7 @@
 
 import tensorflow as tf
 
-import tensorflow_privacy.privacy.optimizers.gaussian_average_query as ph
+import privacy.optimizers.gaussian_average_query as ph
 
 
 class DPAdamOptimizer(tf.train.AdamOptimizer):
diff --git a/privacy/optimizers/dp_gradient_descent.py b/privacy/optimizers/dp_gradient_descent.py
@@ -20,7 +20,7 @@
 
 import tensorflow as tf
 
-import tensorflow_privacy.privacy.optimizers.gaussian_average_query as ph
+import privacy.optimizers.gaussian_average_query as ph
 
 
 class DPGradientDescentOptimizer(tf.train.GradientDescentOptimizer):
diff --git a/privacy/optimizers/dp_optimizer_test.py b/privacy/optimizers/dp_optimizer_test.py
@@ -22,8 +22,8 @@
 import numpy as np
 import tensorflow as tf
 
-from tensorflow_privacy.privacy.optimizers import dp_adam
-from tensorflow_privacy.privacy.optimizers import dp_gradient_descent
+from privacy.optimizers import dp_adam
+from privacy.optimizers import dp_gradient_descent
 
 
 def loss(val0, val1):
diff --git a/privacy/optimizers/gaussian_average_query.py b/privacy/optimizers/gaussian_average_query.py
@@ -23,7 +23,7 @@
 
 import tensorflow as tf
 
-from tensorflow_privacy.privacy.optimizers import private_queries
+from privacy.optimizers import private_queries
 
 
 class GaussianAverageQuery(private_queries.PrivateAverageQuery):
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1 @@
+tensorflow
