Fix copybara to limit some transformations to the beginning of the line.

PiperOrigin-RevId: 233151293

privacy/analysis/compute_dp_sgd_privacy.py

@@ -37,8 +37,8 @@ import math
 from absl import app
 from absl import flags
 
-from tensorflow_privacy.privacy.analysis.rdp_accountant import compute_rdp
-from tensorflow_privacy.privacy.analysis.rdp_accountant import get_privacy_spent
+from privacy.analysis.rdp_accountant import compute_rdp
+from privacy.analysis.rdp_accountant import get_privacy_spent
 
 FLAGS = flags.FLAGS
 

privacy/analysis/privacy_ledger.py

@@ -23,8 +23,8 @@ import collections
 
 import tensorflow as tf
 
-from tensorflow_privacy.privacy.analysis import tensor_buffer
-from tensorflow_privacy.privacy.optimizers import dp_query
+from privacy.analysis import tensor_buffer
+from privacy.optimizers import dp_query
 
 nest = tf.contrib.framework.nest
 

privacy/analysis/privacy_ledger_test.py

@@ -20,10 +20,10 @@ from __future__ import print_function
 
 import tensorflow as tf
 
-from tensorflow_privacy.privacy.analysis import privacy_ledger
-from tensorflow_privacy.privacy.optimizers import gaussian_query
-from tensorflow_privacy.privacy.optimizers import nested_query
-from tensorflow_privacy.privacy.optimizers import test_utils
+from privacy.analysis import privacy_ledger
+from privacy.optimizers import gaussian_query
+from privacy.optimizers import nested_query
+from privacy.optimizers import test_utils
 
 tf.enable_eager_execution()
 

privacy/analysis/rdp_accountant_test.py

@@ -25,7 +25,7 @@ from absl.testing import parameterized
 import mpmath as mp
 import numpy as np
 
-from tensorflow_privacy.privacy.analysis import rdp_accountant
+from privacy.analysis import rdp_accountant
 
 
 class TestGaussianMoments(parameterized.TestCase):

privacy/analysis/tensor_buffer_test.py

@@ -19,7 +19,7 @@ from __future__ import print_function
 
 import tensorflow as tf
 
-from tensorflow_privacy.privacy.analysis import tensor_buffer
+from privacy.analysis import tensor_buffer
 
 tf.enable_eager_execution()
 

privacy/optimizers/dp_optimizer.py

@@ -19,8 +19,8 @@ from __future__ import print_function
 
 import tensorflow as tf
 
-from tensorflow_privacy.privacy.analysis import privacy_ledger
-from tensorflow_privacy.privacy.optimizers import gaussian_query
+from privacy.analysis import privacy_ledger
+from privacy.optimizers import gaussian_query
 
 
 def make_optimizer_class(cls):

privacy/optimizers/dp_optimizer_test.py

@@ -22,9 +22,9 @@ import mock
 import numpy as np
 import tensorflow as tf
 
-from tensorflow_privacy.privacy.analysis import privacy_ledger
-from tensorflow_privacy.privacy.optimizers import dp_optimizer
-from tensorflow_privacy.privacy.optimizers import gaussian_query
+from privacy.analysis import privacy_ledger
+from privacy.optimizers import dp_optimizer
+from privacy.optimizers import gaussian_query
 
 
 def loss(val0, val1):

privacy/optimizers/gaussian_query.py

@@ -23,7 +23,7 @@ import collections
 
 import tensorflow as tf
 
-from tensorflow_privacy.privacy.optimizers import dp_query
+from privacy.optimizers import dp_query
 
 nest = tf.contrib.framework.nest
 

privacy/optimizers/gaussian_query_test.py

@@ -22,8 +22,8 @@ from absl.testing import parameterized
 import numpy as np
 import tensorflow as tf
 
-from tensorflow_privacy.privacy.optimizers import gaussian_query
-from tensorflow_privacy.privacy.optimizers import test_utils
+from privacy.optimizers import gaussian_query
+from privacy.optimizers import test_utils
 
 
 class GaussianQueryTest(tf.test.TestCase, parameterized.TestCase):

privacy/optimizers/nested_query.py

@@ -22,7 +22,7 @@ from __future__ import print_function
 
 import tensorflow as tf
 
-from tensorflow_privacy.privacy.optimizers import dp_query
+from privacy.optimizers import dp_query
 
 nest = tf.contrib.framework.nest
 

privacy/optimizers/nested_query_test.py

@@ -23,9 +23,9 @@ from absl.testing import parameterized
 import numpy as np
 import tensorflow as tf
 
-from tensorflow_privacy.privacy.optimizers import gaussian_query
-from tensorflow_privacy.privacy.optimizers import nested_query
-from tensorflow_privacy.privacy.optimizers import test_utils
+from privacy.optimizers import gaussian_query
+from privacy.optimizers import nested_query
+from privacy.optimizers import test_utils
 
 nest = tf.contrib.framework.nest
 

privacy/optimizers/no_privacy_query.py

@@ -19,7 +19,7 @@ from __future__ import print_function
 
 import tensorflow as tf
 
-from tensorflow_privacy.privacy.optimizers import dp_query
+from privacy.optimizers import dp_query
 
 nest = tf.contrib.framework.nest
 

privacy/optimizers/no_privacy_query_test.py

@@ -21,8 +21,8 @@ from __future__ import print_function
 from absl.testing import parameterized
 import tensorflow as tf
 
-from tensorflow_privacy.privacy.optimizers import no_privacy_query
-from tensorflow_privacy.privacy.optimizers import test_utils
+from privacy.optimizers import no_privacy_query
+from privacy.optimizers import test_utils
 
 
 class NoPrivacyQueryTest(tf.test.TestCase, parameterized.TestCase):

tutorials/mnist_dpsgd_tutorial.py

@@ -21,9 +21,9 @@ from __future__ import print_function
 import numpy as np
 import tensorflow as tf
 
-from tensorflow_privacy.privacy.analysis.rdp_accountant import compute_rdp
-from tensorflow_privacy.privacy.analysis.rdp_accountant import get_privacy_spent
-from tensorflow_privacy.privacy.optimizers import dp_optimizer
+from privacy.analysis.rdp_accountant import compute_rdp
+from privacy.analysis.rdp_accountant import get_privacy_spent
+from privacy.optimizers import dp_optimizer
 
 tf.flags.DEFINE_boolean('dpsgd', True, 'If True, train with DP-SGD. If False, '
                         'train with vanilla SGD.')