Use from tensorflow_privacy import v1 so v1 is visible to the api-generator.

PiperOrigin-RevId: 393866702
This commit is contained in:
Mark Daoust 2021-08-30 15:27:51 -07:00 committed by A. Unique TensorFlower
parent 5edea5863c
commit 54f1887b10

View file

@ -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