diff --git a/tensorflow_federated/python/aggregators/BUILD b/tensorflow_federated/python/aggregators/BUILD index 9ab92f04a753a4a66683f0285b06ed4d6e65d52e..d0035b427fe8429aba871d1c125e933a462e74cd 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 a120a46a964fddc05311cbf60ea7d1ea0832cba6..c2f2107ac17e7bfd010b1466fda5d49bbe5fdba4 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