From 1a959eec34614a5ab79e84e60788840b532c3a9c Mon Sep 17 00:00:00 2001 From: Steve Chien Date: Fri, 17 Jul 2020 09:07:05 -0700 Subject: [PATCH] Fix import of tutorials common module. PiperOrigin-RevId: 321789833 --- tutorials/mnist_dpsgd_tutorial.py | 2 +- tutorials/mnist_dpsgd_tutorial_common.py | 1 - tutorials/mnist_dpsgd_tutorial_tpu.py | 2 +- 3 files changed, 2 insertions(+), 3 deletions(-) diff --git a/tutorials/mnist_dpsgd_tutorial.py b/tutorials/mnist_dpsgd_tutorial.py index e9cf3f8..51b0c40 100644 --- a/tutorials/mnist_dpsgd_tutorial.py +++ b/tutorials/mnist_dpsgd_tutorial.py @@ -28,7 +28,7 @@ import tensorflow.compat.v1 as tf from tensorflow_privacy.privacy.analysis import compute_dp_sgd_privacy_lib from tensorflow_privacy.privacy.optimizers import dp_optimizer -from tensorflow_privacy.tutorials import mnist_dpsgd_tutorial_common as common +import mnist_dpsgd_tutorial_common as common flags.DEFINE_boolean( 'dpsgd', True, 'If True, train with DP-SGD. If False, ' diff --git a/tutorials/mnist_dpsgd_tutorial_common.py b/tutorials/mnist_dpsgd_tutorial_common.py index 656ad54..4784939 100644 --- a/tutorials/mnist_dpsgd_tutorial_common.py +++ b/tutorials/mnist_dpsgd_tutorial_common.py @@ -16,7 +16,6 @@ # These are not necessary in a Python 3-only module. from __future__ import absolute_import from __future__ import division -from __future__ import google_type_annotations from __future__ import print_function import tensorflow.compat.v1 as tf diff --git a/tutorials/mnist_dpsgd_tutorial_tpu.py b/tutorials/mnist_dpsgd_tutorial_tpu.py index e9def32..0df6f40 100644 --- a/tutorials/mnist_dpsgd_tutorial_tpu.py +++ b/tutorials/mnist_dpsgd_tutorial_tpu.py @@ -29,7 +29,7 @@ import tensorflow.compat.v1 as tf from tensorflow_privacy.privacy.analysis import compute_dp_sgd_privacy_lib from tensorflow_privacy.privacy.optimizers import dp_optimizer -from tensorflow_privacy.tutorials import mnist_dpsgd_tutorial_common as common +import mnist_dpsgd_tutorial_common as common flags.DEFINE_boolean( 'dpsgd', True, 'If True, train with DP-SGD. If False, '