From a7301afa0f389b4a29da2883ac8f15a561d7b515 Mon Sep 17 00:00:00 2001
From: Jakub Konecny <konkey@google.com>
Date: Tue, 20 Oct 2020 10:37:44 -0700
Subject: [PATCH] Adds `DifferentialyPrivateSumFactory` to the public API.
PiperOrigin-RevId: 338088240
---
tensorflow_federated/python/aggregators/BUILD | 1 +
tensorflow_federated/python/aggregators/__init__.py | 1 +
2 files changed, 2 insertions(+)
diff --git a/tensorflow_federated/python/aggregators/BUILD b/tensorflow_federated/python/aggregators/BUILD
index 9ab92f04a..d0035b427 100644
--- a/tensorflow_federated/python/aggregators/BUILD
+++ b/tensorflow_federated/python/aggregators/BUILD
@@ -18,6 +18,7 @@ py_library(
srcs_version = "PY3",
visibility = ["//tensorflow_federated:__pkg__"],
deps = [
+ ":dp_factory",
":factory",
":mean_factory",
":quantile_estimation",
diff --git a/tensorflow_federated/python/aggregators/__init__.py b/tensorflow_federated/python/aggregators/__init__.py
index a120a46a9..c2f2107ac 100644
--- a/tensorflow_federated/python/aggregators/__init__.py
+++ b/tensorflow_federated/python/aggregators/__init__.py
@@ -13,6 +13,7 @@
# limitations under the License.
"""Libraries for constructing federated aggregation."""
+from tensorflow_federated.python.aggregators.dp_factory import DifferentiallyPrivateFactory
from tensorflow_federated.python.aggregators.factory import AggregationProcessFactory
from tensorflow_federated.python.aggregators.mean_factory import MeanFactory
from tensorflow_federated.python.aggregators.quantile_estimation import PrivateQuantileEstimationProcess
--
GitLab