diff --git a/tensorflow_privacy/BUILD b/tensorflow_privacy/BUILD index a759c97..47d8ad8 100644 --- a/tensorflow_privacy/BUILD +++ b/tensorflow_privacy/BUILD @@ -22,6 +22,7 @@ py_library( "//tensorflow_privacy/privacy/dp_query", "//tensorflow_privacy/privacy/dp_query:discrete_gaussian_query", "//tensorflow_privacy/privacy/dp_query:distributed_discrete_gaussian_query", + "//tensorflow_privacy/privacy/dp_query:distributed_skellam_query", "//tensorflow_privacy/privacy/dp_query:gaussian_query", "//tensorflow_privacy/privacy/dp_query:nested_query", "//tensorflow_privacy/privacy/dp_query:no_privacy_query", diff --git a/tensorflow_privacy/__init__.py b/tensorflow_privacy/__init__.py index 813c078..caed9b4 100644 --- a/tensorflow_privacy/__init__.py +++ b/tensorflow_privacy/__init__.py @@ -51,6 +51,7 @@ else: from tensorflow_privacy.privacy.dp_query.dp_query import SumAggregationDPQuery from tensorflow_privacy.privacy.dp_query.discrete_gaussian_query import DiscreteGaussianSumQuery from tensorflow_privacy.privacy.dp_query.distributed_discrete_gaussian_query import DistributedDiscreteGaussianSumQuery + from tensorflow_privacy.privacy.dp_query.distributed_skellam_query import DistributedSkellamSumQuery from tensorflow_privacy.privacy.dp_query.gaussian_query import GaussianSumQuery from tensorflow_privacy.privacy.dp_query.nested_query import NestedQuery from tensorflow_privacy.privacy.dp_query.no_privacy_query import NoPrivacyAverageQuery