From c4fea75675431edb85429af405c8441b249acf95 Mon Sep 17 00:00:00 2001 From: Galen Andrew Date: Fri, 10 Jun 2022 14:09:11 -0700 Subject: [PATCH] Remove differential_privacy dp_event classes from TFP top-level symbols. PiperOrigin-RevId: 454238513 --- tensorflow_privacy/BUILD | 1 - tensorflow_privacy/__init__.py | 12 ------------ 2 files changed, 13 deletions(-) diff --git a/tensorflow_privacy/BUILD b/tensorflow_privacy/BUILD index abcd237..48fe304 100644 --- a/tensorflow_privacy/BUILD +++ b/tensorflow_privacy/BUILD @@ -37,7 +37,6 @@ py_library( "//tensorflow_privacy/privacy/optimizers:dp_optimizer_keras", "//tensorflow_privacy/privacy/optimizers:dp_optimizer_keras_vectorized", "//tensorflow_privacy/v1:tensorflow_privacy_v1", - "@com_google_differential_py//python/dp_accounting:dp_event", ], ) diff --git a/tensorflow_privacy/__init__.py b/tensorflow_privacy/__init__.py index 104001d..d25758f 100644 --- a/tensorflow_privacy/__init__.py +++ b/tensorflow_privacy/__init__.py @@ -25,18 +25,6 @@ else: # TensorFlow v1 imports from tensorflow_privacy import v1 - # DpEvents - from com_google_differential_py.python.dp_accounting.dp_event import DpEvent - from com_google_differential_py.python.dp_accounting.dp_event import NoOpDpEvent - from com_google_differential_py.python.dp_accounting.dp_event import NonPrivateDpEvent - from com_google_differential_py.python.dp_accounting.dp_event import UnsupportedDpEvent - from com_google_differential_py.python.dp_accounting.dp_event import GaussianDpEvent - from com_google_differential_py.python.dp_accounting.dp_event import SelfComposedDpEvent - from com_google_differential_py.python.dp_accounting.dp_event import ComposedDpEvent - from com_google_differential_py.python.dp_accounting.dp_event import PoissonSampledDpEvent - from com_google_differential_py.python.dp_accounting.dp_event import SampledWithReplacementDpEvent - from com_google_differential_py.python.dp_accounting.dp_event import SampledWithoutReplacementDpEvent - # Analysis from tensorflow_privacy.privacy.analysis.compute_dp_sgd_privacy_lib import compute_dp_sgd_privacy from tensorflow_privacy.privacy.analysis.rdp_accountant import compute_heterogeneous_rdp