From f3672adf804ee8e00679417964081e90ec743f96 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Fri, 6 Mar 2020 11:28:38 -0800 Subject: [PATCH] Explicitly disable Tensorflow v2 behaviors for all TF1.x binaries and tests PiperOrigin-RevId: 299397784 --- tensorflow_privacy/privacy/dp_query/BUILD | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tensorflow_privacy/privacy/dp_query/BUILD b/tensorflow_privacy/privacy/dp_query/BUILD index 2931140..7118089 100644 --- a/tensorflow_privacy/privacy/dp_query/BUILD +++ b/tensorflow_privacy/privacy/dp_query/BUILD @@ -30,6 +30,7 @@ py_test( deps = [ ":gaussian_query", ":test_utils", + "//learning/brain/public:disable_tf2", # build_cleaner: keep; go/disable_tf2 "//third_party/py/absl/testing:parameterized", "//third_party/py/numpy", "//third_party/py/six", @@ -55,6 +56,7 @@ py_test( deps = [ ":no_privacy_query", ":test_utils", + "//learning/brain/public:disable_tf2", # build_cleaner: keep; go/disable_tf2 "//third_party/py/absl/testing:parameterized", "//third_party/py/tensorflow", ], @@ -79,6 +81,7 @@ py_test( ":gaussian_query", ":normalized_query", ":test_utils", + "//learning/brain/public:disable_tf2", # build_cleaner: keep; go/disable_tf2 "//third_party/py/tensorflow", ], ) @@ -102,6 +105,7 @@ py_test( ":gaussian_query", ":nested_query", ":test_utils", + "//learning/brain/public:disable_tf2", # build_cleaner: keep; go/disable_tf2 "//third_party/py/absl/testing:parameterized", "//third_party/py/distutils", "//third_party/py/numpy", @@ -127,6 +131,7 @@ py_test( deps = [ ":quantile_adaptive_clip_sum_query", ":test_utils", + "//learning/brain/public:disable_tf2", # build_cleaner: keep; go/disable_tf2 "//third_party/py/absl/testing:parameterized", "//third_party/py/numpy", "//third_party/py/tensorflow",