forked from 626_privacy/tensorflow_privacy
Use from tensorflow_privacy import v1
so v1 is visible to the api-generator.
PiperOrigin-RevId: 393866702
This commit is contained in:
parent
5edea5863c
commit
54f1887b10
1 changed files with 1 additions and 1 deletions
|
@ -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
|
||||
|
|
Loading…
Reference in a new issue