From 54f1887b10fdbe518be1dc5d164cbda979d904d7 Mon Sep 17 00:00:00 2001 From: Mark Daoust Date: Mon, 30 Aug 2021 15:27:51 -0700 Subject: [PATCH] Use `from tensorflow_privacy import v1` so v1 is visible to the api-generator. PiperOrigin-RevId: 393866702 --- tensorflow_privacy/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow_privacy/__init__.py b/tensorflow_privacy/__init__.py index f855eba..84839e4 100644 --- a/tensorflow_privacy/__init__.py +++ b/tensorflow_privacy/__init__.py @@ -27,7 +27,7 @@ if hasattr(sys, 'skip_tf_privacy_import'): # Useful for standalone scripts. pass else: # TensorFlow v1 imports - import tensorflow_privacy.v1 + from tensorflow_privacy import v1 # Analysis from tensorflow_privacy.privacy.analysis.compute_dp_sgd_privacy_lib import compute_dp_sgd_privacy