Explicitly disable Tensorflow v2 behaviors for all TF1.x binaries and tests
PiperOrigin-RevId: 299397784
This commit is contained in:
parent
b0df24ef25
commit
f3672adf80
1 changed files with 5 additions and 0 deletions
|
@ -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",
|
||||||
|
|
Loading…
Reference in a new issue