diff --git a/tensorflow_privacy/__init__.py b/tensorflow_privacy/__init__.py index 648e98a..4942ca6 100644 --- a/tensorflow_privacy/__init__.py +++ b/tensorflow_privacy/__init__.py @@ -37,6 +37,7 @@ else: from tensorflow_privacy.privacy.dp_query.no_privacy_query import NoPrivacySumQuery from tensorflow_privacy.privacy.dp_query.normalized_query import NormalizedQuery from tensorflow_privacy.privacy.dp_query.quantile_estimator_query import QuantileEstimatorQuery + from tensorflow_privacy.privacy.dp_query.quantile_estimator_query import NoPrivacyQuantileEstimatorQuery from tensorflow_privacy.privacy.dp_query.quantile_adaptive_clip_sum_query import QuantileAdaptiveClipSumQuery from tensorflow_privacy.privacy.dp_query.quantile_adaptive_clip_sum_query import QuantileAdaptiveClipAverageQuery