Explicitly disable Tensorflow v2 behaviors for all TF1.x binaries and tests

PiperOrigin-RevId: 299397784
This commit is contained in:
A. Unique TensorFlower 2020-03-06 11:28:38 -08:00
parent b0df24ef25
commit f3672adf80

View file

@ -30,6 +30,7 @@ py_test(
deps = [ deps = [
":gaussian_query", ":gaussian_query",
":test_utils", ":test_utils",
"//learning/brain/public:disable_tf2", # build_cleaner: keep; go/disable_tf2
"//third_party/py/absl/testing:parameterized", "//third_party/py/absl/testing:parameterized",
"//third_party/py/numpy", "//third_party/py/numpy",
"//third_party/py/six", "//third_party/py/six",
@ -55,6 +56,7 @@ py_test(
deps = [ deps = [
":no_privacy_query", ":no_privacy_query",
":test_utils", ":test_utils",
"//learning/brain/public:disable_tf2", # build_cleaner: keep; go/disable_tf2
"//third_party/py/absl/testing:parameterized", "//third_party/py/absl/testing:parameterized",
"//third_party/py/tensorflow", "//third_party/py/tensorflow",
], ],
@ -79,6 +81,7 @@ py_test(
":gaussian_query", ":gaussian_query",
":normalized_query", ":normalized_query",
":test_utils", ":test_utils",
"//learning/brain/public:disable_tf2", # build_cleaner: keep; go/disable_tf2
"//third_party/py/tensorflow", "//third_party/py/tensorflow",
], ],
) )
@ -102,6 +105,7 @@ py_test(
":gaussian_query", ":gaussian_query",
":nested_query", ":nested_query",
":test_utils", ":test_utils",
"//learning/brain/public:disable_tf2", # build_cleaner: keep; go/disable_tf2
"//third_party/py/absl/testing:parameterized", "//third_party/py/absl/testing:parameterized",
"//third_party/py/distutils", "//third_party/py/distutils",
"//third_party/py/numpy", "//third_party/py/numpy",
@ -127,6 +131,7 @@ py_test(
deps = [ deps = [
":quantile_adaptive_clip_sum_query", ":quantile_adaptive_clip_sum_query",
":test_utils", ":test_utils",
"//learning/brain/public:disable_tf2", # build_cleaner: keep; go/disable_tf2
"//third_party/py/absl/testing:parameterized", "//third_party/py/absl/testing:parameterized",
"//third_party/py/numpy", "//third_party/py/numpy",
"//third_party/py/tensorflow", "//third_party/py/tensorflow",