n/a

PiperOrigin-RevId: 247680554

tensorflow/compiler/aot/tests/BUILD

@@ -36,6 +36,7 @@ py_binary(
     name = "make_test_graphs",
     testonly = 1,
     srcs = ["make_test_graphs.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow/core:protos_all_py",

tensorflow/examples/learn/BUILD

@@ -12,6 +12,7 @@ exports_files(["LICENSE"])
 py_binary(
     name = "iris_custom_decay_dnn",
     srcs = ["iris_custom_decay_dnn.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = ["//tensorflow:tensorflow_py"],
 )
@@ -19,6 +20,7 @@ py_binary(
 py_binary(
     name = "iris_custom_model",
     srcs = ["iris_custom_model.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = ["//tensorflow:tensorflow_py"],
 )

tensorflow/examples/tutorials/mnist/BUILD

@@ -50,6 +50,7 @@ py_binary(
     srcs = [
         "fully_connected_feed.py",
     ],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = ["optonly"],
     deps = [
@@ -64,6 +65,7 @@ py_binary(
     srcs = [
         "mnist_with_summaries.py",
     ],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":input_data",
@@ -82,6 +84,7 @@ py_binary(
     srcs = [
         "mnist_softmax_xla.py",
     ],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":input_data",
@@ -100,6 +103,7 @@ py_test(
         "--max_steps=10",
     ],
     main = "fully_connected_feed.py",
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":input_data",
@@ -120,6 +124,7 @@ py_test(
         "--learning_rate=0.00",
     ],
     main = "mnist_with_summaries.py",
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = ["notsan"],  # http://b/29184009
     deps = [

tensorflow/lite/experimental/examples/lstm/BUILD

@@ -36,6 +36,7 @@ py_test(
     name = "unidirectional_sequence_lstm_test",
     size = "large",
     srcs = ["unidirectional_sequence_lstm_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -58,6 +59,7 @@ py_test(
     name = "unidirectional_sequence_rnn_test",
     size = "large",
     srcs = ["unidirectional_sequence_rnn_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -80,6 +82,7 @@ py_test(
     name = "bidirectional_sequence_lstm_test",
     size = "large",
     srcs = ["bidirectional_sequence_lstm_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -102,6 +105,7 @@ py_test(
     name = "bidirectional_sequence_rnn_test",
     size = "large",
     srcs = ["bidirectional_sequence_rnn_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",

tensorflow/lite/python/BUILD

@@ -21,6 +21,7 @@ py_test(
     name = "interpreter_test",
     srcs = ["interpreter_test.py"],
     data = ["//tensorflow/lite/python/testdata:interpreter_test_data"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_windows",
@@ -37,6 +38,7 @@ py_test(
 py_binary(
     name = "tflite_convert",
     srcs = ["tflite_convert.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     visibility = ["//visibility:public"],
     deps = [":tflite_convert_main_lib"],
@@ -86,6 +88,7 @@ py_test(
     name = "lite_test",
     srcs = ["lite_test.py"],
     data = ["@tflite_mobilenet_ssd_quant_protobuf//:tflite_graph.pb"],
+    python_version = "PY2",
     shard_count = 4,
     srcs_version = "PY2AND3",
     tags = [
@@ -101,6 +104,7 @@ py_test(
 py_test(
     name = "lite_v2_test",
     srcs = ["lite_v2_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_windows",
@@ -115,6 +119,7 @@ py_test(
 py_test(
     name = "lite_flex_test",
     srcs = ["lite_flex_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         # TODO(b/111881877): Enable in oss after resolving op registry issues.
@@ -145,6 +150,7 @@ py_library(
 py_test(
     name = "util_test",
     srcs = ["util_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_windows",
@@ -210,6 +216,7 @@ py_library(
 py_test(
     name = "convert_test",
     srcs = ["convert_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":convert",
@@ -241,6 +248,7 @@ py_library(
 py_test(
     name = "convert_saved_model_test",
     srcs = ["convert_saved_model_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_windows",
@@ -262,6 +270,7 @@ py_test(
 py_binary(
     name = "create_custom_op",
     srcs = ["create_custom_op.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     visibility = ["//visibility:public"],
     deps = [

tensorflow/lite/python/optimize/BUILD

@@ -58,6 +58,7 @@ py_test(
         ":test_data",
         "//tensorflow/lite:testdata/multi_add.bin",
     ],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = ["no_oss"],
     deps = [

tensorflow/lite/toco/python/BUILD

@@ -61,6 +61,7 @@ tf_py_wrap_cc(
 py_binary(
     name = "toco_from_protos",
     srcs = ["toco_from_protos.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":tensorflow_wrap_toco",

tensorflow/python/data/benchmarks/BUILD

@@ -9,6 +9,7 @@ load("//tensorflow:tensorflow.bzl", "py_test")
 py_test(
     name = "meta_benchmark",
     srcs = ["meta_benchmark.py"],
+    python_version = "PY2",
     deps = [
         "//tensorflow/python:client_testlib",
         "//tensorflow/python:session",
@@ -31,6 +32,7 @@ py_library(
 py_test(
     name = "batch_benchmark",
     srcs = ["batch_benchmark.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":benchmark_base",
@@ -43,6 +45,7 @@ py_test(
 py_test(
     name = "filter_benchmark",
     srcs = ["filter_benchmark.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":benchmark_base",
@@ -53,6 +56,7 @@ py_test(
 py_test(
     name = "from_tensor_slices_benchmark",
     srcs = ["from_tensor_slices_benchmark.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":benchmark_base",
@@ -64,6 +68,7 @@ py_test(
 py_test(
     name = "list_files_benchmark",
     srcs = ["list_files_benchmark.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":benchmark_base",
@@ -79,6 +84,7 @@ py_test(
 py_test(
     name = "map_benchmark",
     srcs = ["map_benchmark.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":benchmark_base",
@@ -89,6 +95,7 @@ py_test(
 py_test(
     name = "range_benchmark",
     srcs = ["range_benchmark.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":benchmark_base",

tensorflow/python/data/experimental/kernel_tests/BUILD

@@ -11,6 +11,7 @@ py_test(
     name = "bucket_by_sequence_length_test",
     size = "medium",
     srcs = ["bucket_by_sequence_length_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow/python:array_ops",
@@ -30,6 +31,7 @@ py_test(
 py_test(
     name = "cardinality_test",
     srcs = ["cardinality_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow/python/data/experimental/ops:cardinality",
@@ -64,6 +66,7 @@ py_test(
     name = "counter_test",
     size = "small",
     srcs = ["counter_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow/python:client_testlib",
@@ -77,6 +80,7 @@ py_test(
     name = "csv_dataset_test",
     size = "medium",
     srcs = ["csv_dataset_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = ["no_pip"],
     deps = [
@@ -97,6 +101,7 @@ py_test(
 py_test(
     name = "dense_to_sparse_batch_test",
     srcs = ["dense_to_sparse_batch_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow/python:array_ops",
@@ -114,6 +119,7 @@ py_test(
     name = "directed_interleave_dataset_test",
     size = "medium",
     srcs = ["directed_interleave_dataset_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow/python:client_testlib",
@@ -130,6 +136,7 @@ py_test(
     name = "auto_shard_dataset_test",
     size = "medium",
     srcs = ["auto_shard_dataset_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = ["no_pip"],
     deps = [
@@ -148,6 +155,7 @@ py_test(
     name = "get_single_element_test",
     size = "small",
     srcs = ["get_single_element_test.py"],
+    python_version = "PY2",
     deps = [
         "//tensorflow/python:array_ops",
         "//tensorflow/python:client_testlib",
@@ -167,6 +175,7 @@ py_test(
     name = "group_by_reducer_test",
     size = "medium",
     srcs = ["group_by_reducer_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow/python:array_ops",
@@ -188,6 +197,7 @@ py_test(
     name = "group_by_window_test",
     size = "medium",
     srcs = ["group_by_window_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow/python:array_ops",
@@ -209,6 +219,7 @@ py_test(
 py_test(
     name = "ignore_errors_test",
     srcs = ["ignore_errors_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow/python:array_ops",
@@ -227,6 +238,7 @@ py_test(
     name = "make_batched_features_dataset_test",
     size = "medium",
     srcs = ["make_batched_features_dataset_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = ["no_pip"],
     deps = [
@@ -248,6 +260,7 @@ py_test(
     name = "make_csv_dataset_test",
     size = "medium",
     srcs = ["make_csv_dataset_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = ["no_pip"],
     deps = [
@@ -267,6 +280,7 @@ py_test(
     name = "make_tf_record_dataset_test",
     size = "medium",
     srcs = ["make_tf_record_dataset_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = ["no_pip"],
     deps = [
@@ -284,6 +298,7 @@ py_test(
     name = "map_and_batch_test",
     size = "medium",
     srcs = ["map_and_batch_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow/python:array_ops",
@@ -307,6 +322,7 @@ py_test(
     name = "map_defun_op_test",
     size = "small",
     srcs = ["map_defun_op_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = ["no_pip"],
     deps = [
@@ -332,6 +348,7 @@ py_test(
     name = "matching_files_test",
     size = "small",
     srcs = ["matching_files_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = ["no_pip"],
     deps = [
@@ -351,6 +368,7 @@ py_test(
     name = "override_threadpool_test",
     size = "small",
     srcs = ["override_threadpool_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = ["no_pip"],
     deps = [
@@ -371,6 +389,7 @@ py_test(
     name = "parallel_interleave_test",
     size = "medium",
     srcs = ["parallel_interleave_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = ["no_pip"],
     deps = [
@@ -393,6 +412,7 @@ py_test(
     name = "parse_example_dataset_test",
     size = "small",
     srcs = ["parse_example_dataset_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow/core:protos_all_py",
@@ -455,6 +475,7 @@ py_test(
     name = "rebatch_dataset_test",
     size = "small",
     srcs = ["rebatch_dataset_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow/python:client_testlib",
@@ -472,6 +493,7 @@ py_test(
     name = "rejection_resample_test",
     size = "medium",
     srcs = ["rejection_resample_test.py"],
+    python_version = "PY2",
     shard_count = 2,
     srcs_version = "PY2AND3",
     tags = [
@@ -499,6 +521,7 @@ py_test(
     name = "restructured_dataset_test",
     size = "medium",
     srcs = ["restructured_dataset_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow/python:array_ops",
@@ -515,6 +538,7 @@ py_test(
     name = "scan_test",
     size = "small",
     srcs = ["scan_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = ["no_pip"],
     deps = [
@@ -539,6 +563,7 @@ py_test(
     name = "shuffle_and_repeat_test",
     size = "medium",
     srcs = ["shuffle_and_repeat_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_pip",
@@ -558,6 +583,7 @@ py_test(
 py_test(
     name = "sleep_test",
     srcs = ["sleep_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow/python:client_testlib",
@@ -590,6 +616,7 @@ py_test(
     name = "sql_dataset_test",
     size = "medium",
     srcs = ["sql_dataset_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = ["no_pip"],
     deps = [
@@ -603,6 +630,7 @@ py_test(
 py_test(
     name = "snapshot_test",
     srcs = ["snapshot_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":reader_dataset_ops_test_base",
@@ -622,6 +650,7 @@ py_test(
     name = "stats_dataset_ops_test",
     size = "large",
     srcs = ["stats_dataset_ops_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_pip",
@@ -660,6 +689,7 @@ py_test(
     name = "take_while_test",
     size = "small",
     srcs = ["take_while_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow/python:array_ops",
@@ -682,6 +712,7 @@ py_test(
     name = "tf_record_writer_test",
     size = "small",
     srcs = ["tf_record_writer_test.py"],
+    python_version = "PY2",
     deps = [
         "//tensorflow/python:array_ops",
         "//tensorflow/python:client_testlib",
@@ -699,6 +730,7 @@ py_test(
     name = "unbatch_test",
     size = "medium",
     srcs = ["unbatch_test.py"],
+    python_version = "PY2",
     deps = [
         "//tensorflow/python:array_ops",
         "//tensorflow/python:client_testlib",
@@ -725,6 +757,7 @@ py_test(
     name = "unique_test",
     size = "small",
     srcs = ["unique_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = ["no_pip"],
     deps = [
@@ -741,6 +774,7 @@ py_test(
 py_test(
     name = "variant_test",
     srcs = ["variant_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow/python:client_testlib",

tensorflow/python/data/experimental/kernel_tests/serialization/BUILD

@@ -33,6 +33,7 @@ py_test(
     name = "batch_dataset_serialization_test",
     size = "medium",
     srcs = ["batch_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -54,6 +55,7 @@ py_test(
     name = "cache_dataset_serialization_test",
     size = "small",
     srcs = ["cache_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -73,6 +75,7 @@ py_test(
     name = "checkpoint_input_pipeline_hook_test",
     size = "small",
     srcs = ["checkpoint_input_pipeline_hook_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_pip",
@@ -97,6 +100,7 @@ py_test(
     name = "choose_fastest_branch_dataset_serialization_test",
     size = "medium",
     srcs = ["choose_fastest_branch_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -118,6 +122,7 @@ py_test(
     name = "choose_fastest_dataset_serialization_test",
     size = "small",
     srcs = ["choose_fastest_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -136,6 +141,7 @@ py_test(
     name = "concatenate_dataset_serialization_test",
     size = "small",
     srcs = ["concatenate_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -154,6 +160,7 @@ py_test(
     name = "csv_dataset_serialization_test",
     size = "small",
     srcs = ["csv_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -172,6 +179,7 @@ py_test(
     name = "dataset_constructor_serialization_test",
     size = "medium",
     srcs = ["dataset_constructor_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -191,6 +199,7 @@ py_test(
     name = "auto_shard_dataset_serialization_test",
     size = "medium",
     srcs = ["auto_shard_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -214,6 +223,7 @@ py_test(
     name = "filter_dataset_serialization_test",
     size = "medium",
     srcs = ["filter_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -234,6 +244,7 @@ py_test(
     name = "fixed_length_record_dataset_serialization_test",
     size = "medium",
     srcs = ["fixed_length_record_dataset_serialization_test.py"],
+    python_version = "PY2",
     shard_count = 4,
     srcs_version = "PY2AND3",
     tags = [
@@ -253,6 +264,7 @@ py_test(
     name = "flat_map_dataset_serialization_test",
     size = "medium",
     srcs = ["flat_map_dataset_serialization_test.py"],
+    python_version = "PY2",
     tags = [
         "no_oss",
         "no_pip",
@@ -278,6 +290,7 @@ py_test(
     name = "group_by_reducer_serialization_test",
     size = "medium",
     srcs = ["group_by_reducer_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -297,6 +310,7 @@ py_test(
     name = "group_by_window_serialization_test",
     size = "medium",
     srcs = ["group_by_window_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -316,6 +330,7 @@ py_test(
     name = "ignore_errors_serialization_test",
     size = "small",
     srcs = ["ignore_errors_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -336,6 +351,7 @@ py_test(
     name = "interleave_dataset_serialization_test",
     size = "medium",
     srcs = ["interleave_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -357,6 +373,7 @@ py_test(
     name = "map_and_batch_dataset_serialization_test",
     size = "medium",
     srcs = ["map_and_batch_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -376,6 +393,7 @@ py_test(
     name = "map_dataset_serialization_test",
     size = "medium",
     srcs = ["map_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -402,6 +420,7 @@ py_test(
     name = "matching_files_dataset_serialization_test",
     size = "small",
     srcs = ["matching_files_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_windows",
@@ -419,6 +438,7 @@ py_test(
     name = "optimize_dataset_serialization_test",
     size = "small",
     srcs = ["optimize_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -437,6 +457,7 @@ py_test(
     name = "rebatch_dataset_serialization_test",
     size = "small",
     srcs = ["rebatch_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -455,6 +476,7 @@ py_test(
     name = "padded_batch_dataset_serialization_test",
     size = "medium",
     srcs = ["padded_batch_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -475,6 +497,7 @@ py_test(
     name = "parallel_interleave_dataset_serialization_test",
     size = "medium",
     srcs = ["parallel_interleave_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -496,6 +519,7 @@ py_test(
     name = "parallel_map_dataset_serialization_test",
     size = "medium",
     srcs = ["parallel_map_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -521,6 +545,7 @@ py_test(
     name = "parse_example_dataset_serialization_test",
     size = "medium",
     srcs = ["parse_example_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -538,6 +563,7 @@ py_test(
     name = "prefetch_dataset_serialization_test",
     size = "small",
     srcs = ["prefetch_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -555,6 +581,7 @@ py_test(
     name = "range_dataset_serialization_test",
     size = "small",
     srcs = ["range_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -579,6 +606,7 @@ py_test(
     name = "sample_from_datasets_serialization_test",
     size = "medium",
     srcs = ["sample_from_datasets_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -597,6 +625,7 @@ py_test(
     name = "scan_dataset_serialization_test",
     size = "small",
     srcs = ["scan_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -615,6 +644,7 @@ py_test(
     name = "sequence_dataset_serialization_test",
     size = "medium",
     srcs = ["sequence_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -633,6 +663,7 @@ py_test(
     name = "serialization_integration_test",
     size = "small",
     srcs = ["serialization_integration_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -652,6 +683,7 @@ py_test(
     name = "shard_dataset_serialization_test",
     size = "medium",
     srcs = ["shard_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -670,6 +702,7 @@ py_test(
     name = "shuffle_and_repeat_dataset_serialization_test",
     size = "medium",
     srcs = ["shuffle_and_repeat_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -688,6 +721,7 @@ py_test(
     name = "shuffle_dataset_serialization_test",
     size = "medium",
     srcs = ["shuffle_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -708,6 +742,7 @@ py_test(
     name = "sql_dataset_serialization_test",
     size = "small",
     srcs = ["sql_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -728,6 +763,7 @@ py_test(
     name = "stats_dataset_serialization_test",
     size = "medium",
     srcs = ["stats_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -749,6 +785,7 @@ py_test(
     name = "take_while_dataset_serialization_test",
     size = "small",
     srcs = ["take_while_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -768,6 +805,7 @@ py_test(
     name = "textline_dataset_serialization_test",
     size = "medium",
     srcs = ["textline_dataset_serialization_test.py"],
+    python_version = "PY2",
     shard_count = 4,
     srcs_version = "PY2AND3",
     tags = [
@@ -787,6 +825,7 @@ py_test(
     name = "tf_record_dataset_serialization_test",
     size = "medium",
     srcs = ["tf_record_dataset_serialization_test.py"],
+    python_version = "PY2",
     shard_count = 4,
     srcs_version = "PY2AND3",
     tags = [
@@ -806,6 +845,7 @@ py_test(
     name = "unbatch_dataset_serialization_test",
     size = "medium",
     srcs = ["unbatch_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -825,6 +865,7 @@ py_test(
     name = "unique_dataset_serialization_test",
     size = "small",
     srcs = ["unique_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",
@@ -843,6 +884,7 @@ py_test(
     name = "zip_dataset_serialization_test",
     size = "small",
     srcs = ["zip_dataset_serialization_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_oss",

tensorflow/python/debug/BUILD

@@ -116,6 +116,7 @@ py_library(
 py_binary(
     name = "grpc_tensorflow_server",
     srcs = ["lib/grpc_tensorflow_server.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [":grpc_tensorflow_server_lib"],
 )
@@ -381,6 +382,7 @@ py_library(
 py_binary(
     name = "offline_analyzer",
     srcs = ["cli/offline_analyzer.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [":offline_analyzer_lib"],
 )
@@ -412,6 +414,7 @@ py_library(
 py_binary(
     name = "debug_fibonacci",
     srcs = ["examples/debug_fibonacci.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [":debug_fibonacci_lib"],
 )
@@ -431,6 +434,7 @@ py_library(
 py_binary(
     name = "debug_errors",
     srcs = ["examples/debug_errors.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [":debug_errors_lib"],
 )
@@ -449,6 +453,7 @@ py_library(
 py_binary(
     name = "debug_mnist",
     srcs = ["examples/debug_mnist.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [":debug_mnist_lib"],
 )
@@ -467,6 +472,7 @@ py_library(
 py_binary(
     name = "debug_tflearn_iris",
     srcs = ["examples/debug_tflearn_iris.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [":debug_tflearn_iris_lib"],
 )
@@ -485,6 +491,7 @@ py_library(
 py_binary(
     name = "debug_keras",
     srcs = ["examples/debug_keras.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [":debug_keras_lib"],
 )
@@ -504,6 +511,7 @@ py_test(
     name = "common_test",
     size = "small",
     srcs = ["lib/common_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":common",
@@ -518,6 +526,7 @@ py_test(
     name = "debug_graphs_test",
     size = "small",
     srcs = ["lib/debug_graphs_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":debug_graphs",
@@ -530,6 +539,7 @@ py_test(
     name = "debug_data_test",
     size = "small",
     srcs = ["lib/debug_data_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":debug_data",
@@ -565,6 +575,7 @@ py_test(
     name = "debug_utils_test",
     size = "small",
     srcs = ["lib/debug_utils_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":debug_utils",
@@ -584,6 +595,7 @@ py_test(
     name = "source_utils_test",
     size = "small",
     srcs = ["lib/source_utils_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":debug_data",
@@ -610,6 +622,7 @@ py_test(
     name = "source_remote_test",
     size = "small",
     srcs = ["lib/source_remote_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_windows",
@@ -636,6 +649,7 @@ py_test(
     name = "framework_test",
     size = "medium",
     srcs = ["wrappers/framework_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":debug_data",
@@ -660,6 +674,7 @@ py_test(
     name = "profiling_test",
     size = "small",
     srcs = ["lib/profiling_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":profiling",
@@ -673,6 +688,7 @@ py_test(
     name = "curses_ui_test",
     size = "small",
     srcs = ["cli/curses_ui_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_windows",
@@ -693,6 +709,7 @@ py_test(
     name = "readline_ui_test",
     size = "small",
     srcs = ["cli/readline_ui_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":cli_config",
@@ -832,6 +849,7 @@ py_test(
     name = "debugger_cli_common_test",
     size = "small",
     srcs = ["cli/debugger_cli_common_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":debugger_cli_common",
@@ -846,6 +864,7 @@ py_test(
     name = "cli_config_test",
     size = "small",
     srcs = ["cli/cli_config_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":cli_config",
@@ -859,6 +878,7 @@ py_test(
     name = "command_parser_test",
     size = "small",
     srcs = ["cli/command_parser_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":command_parser",
@@ -871,6 +891,7 @@ py_test(
     name = "tensor_format_test",
     size = "small",
     srcs = ["cli/tensor_format_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":cli_test_utils",
@@ -888,6 +909,7 @@ py_test(
     name = "cli_shared_test",
     size = "small",
     srcs = ["cli/cli_shared_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":cli_shared",
@@ -907,6 +929,7 @@ py_test(
     srcs = [
         "cli/evaluator_test.py",
     ],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":debug_data",
@@ -958,6 +981,7 @@ py_test(
     name = "profile_analyzer_cli_test",
     size = "small",
     srcs = ["cli/profile_analyzer_cli_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":debugger_cli_common",
@@ -1061,6 +1085,7 @@ py_test(
     name = "dumping_wrapper_test",
     size = "small",
     srcs = ["wrappers/dumping_wrapper_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":debug_data",
@@ -1083,6 +1108,7 @@ py_test(
     name = "local_cli_wrapper_test",
     size = "small",
     srcs = ["wrappers/local_cli_wrapper_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":cli_shared",
@@ -1110,6 +1136,7 @@ py_test(
     name = "disk_usage_test",
     size = "small",
     srcs = ["wrappers/disk_usage_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":dumping_wrapper",

tensorflow/python/feature_column/BUILD

@@ -238,6 +238,7 @@ tf_py_test(
 py_test(
     name = "sequence_feature_column_integration_test",
     srcs = ["sequence_feature_column_integration_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = ["no_pip"],
     deps = [

tensorflow/python/keras/mixed_precision/experimental/BUILD

@@ -52,6 +52,7 @@ py_test(
     srcs = [
         "policy_test.py",
     ],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":policy",
@@ -78,6 +79,7 @@ py_test(
     name = "autocast_variable_test",
     size = "medium",
     srcs = ["autocast_variable_test.py"],
+    python_version = "PY2",
     deps = [
         ":autocast_variable",
         "//tensorflow/python:client_testlib",
@@ -128,6 +130,7 @@ py_test(
     name = "loss_scale_test",
     size = "medium",
     srcs = ["loss_scale_test.py"],
+    python_version = "PY2",
     deps = [
         ":loss_scale",
         "//tensorflow/python:client_testlib",

tensorflow/python/keras/optimizer_v2/BUILD

@@ -203,6 +203,7 @@ py_test(
     name = "optimizer_v2_test",
     size = "medium",
     srcs = ["optimizer_v2_test.py"],
+    python_version = "PY2",
     shard_count = 8,
     tags = [
         "no_gpu",  # b/127001953

tensorflow/python/ops/losses/BUILD

@@ -42,6 +42,7 @@ py_test(
     name = "util_test",
     size = "small",
     srcs = ["util_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":losses",

tensorflow/python/tools/api/generator/BUILD

@@ -43,6 +43,7 @@ py_test(
         "create_python_api.py",
         "create_python_api_test.py",
     ],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":doc_srcs",
@@ -59,6 +60,7 @@ py_test(
         "--api_name=tensorflow",
     ] + KERAS_API_INIT_FILES + KERAS_API_INIT_FILES_V1 + TENSORFLOW_API_INIT_FILES + TENSORFLOW_API_INIT_FILES_V1,
     main = "doc_srcs_test.py",
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":doc_srcs",
@@ -74,6 +76,7 @@ py_test(
         "api_init_files.bzl",
         "api_init_files_v1.bzl",
     ],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = ["no_pip"],
     deps = [

tensorflow/tools/compatibility/BUILD

@@ -28,6 +28,7 @@ py_library(
 py_test(
     name = "ast_edits_test",
     srcs = ["ast_edits_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":ast_edits",
@@ -40,6 +41,7 @@ py_test(
 py_binary(
     name = "tf_upgrade",
     srcs = ["tf_upgrade.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [":tf_upgrade_lib"],
 )
@@ -54,6 +56,7 @@ py_library(
 py_test(
     name = "tf_upgrade_test",
     srcs = ["tf_upgrade_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     tags = [
         "no_pip",
@@ -104,6 +107,7 @@ py_binary(
     name = "tf_upgrade_v2",
     srcs = ["tf_upgrade_v2_main.py"],
     main = "tf_upgrade_v2_main.py",
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":ast_edits",
@@ -115,6 +119,7 @@ py_binary(
 py_test(
     name = "tf_upgrade_v2_test",
     srcs = ["tf_upgrade_v2_test.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":tf_upgrade_v2_lib",
@@ -158,6 +163,7 @@ py_test(
     name = "test_file_v1_0",
     size = "small",
     srcs = ["test_file_v1_0.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow:tensorflow_py",
@@ -183,6 +189,7 @@ py_test(
     name = "test_file_v1_12",
     size = "small",
     srcs = ["testdata/test_file_v1_12.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow:tensorflow_py",
@@ -193,6 +200,7 @@ py_test(
     name = "test_file_v2_0",
     size = "small",
     srcs = ["test_file_v2_0.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         "//tensorflow:tensorflow_py",

tensorflow/tools/git/BUILD

@@ -9,5 +9,6 @@ licenses(["notice"])  # Apache 2.0
 py_binary(
     name = "gen_git_source",
     srcs = ["gen_git_source.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
 )

tensorflow/tools/tensorflow_builder/BUILD

@@ -15,6 +15,7 @@ py_binary(
     data = [
         "//tensorflow/tools/tensorflow_builder/data/golden:cuda_cc_golden",
     ],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         ":cuda_compute_capability",
@@ -26,6 +27,7 @@ py_binary(
 py_binary(
     name = "cuda_compute_capability",
     srcs = ["data/cuda_compute_capability.py"],
+    python_version = "PY2",
     srcs_version = "PY2AND3",
     deps = [
         "@absl_py//absl:app",