Disable tests relying on TF1 behavior

As part of the conversion to TF2, this ptch disables all tests that rely
on TF1 behavior by calling disable_v2_behavior(). This is so that the
functionality can be incrementally ported to TF2 in separate patches
without having to worry about dependencies. The tests will be
selectively reenabled as the functionality is ported.

Pull Request: google#278

gematria/granite/python/BUILD.bazel

@@ -24,6 +24,9 @@ gematria_py_test(
     timeout = "moderate",
     srcs = ["gnn_model_base_test.py"],
     shard_count = 15,
+    tags = [
+        "manual",
+    ],
     deps = [
         ":gnn_model_base",
         "//gematria/model/python:options",
@@ -88,6 +91,9 @@ gematria_py_test(
     size = "small",
     timeout = "moderate",
     srcs = ["graph_builder_model_base_test.py"],
+    tags = [
+        "manual",
+    ],
     deps = [
         ":gnn_model_base",
         ":graph_builder",
@@ -121,6 +127,9 @@ gematria_py_test(
     timeout = "moderate",
     srcs = ["rnn_token_model_test.py"],
     shard_count = 18,
+    tags = [
+        "manual",
+    ],
     deps = [
         ":rnn_token_model",
         "//gematria/basic_block/python:tokens",
@@ -166,6 +175,9 @@ gematria_py_test(
     timeout = "moderate",
     srcs = ["token_graph_builder_model_test.py"],
     shard_count = 26,
+    tags = [
+        "manual",
+    ],
     deps = [
         ":token_graph_builder_model",
         "//gematria/basic_block/python:tokens",

gematria/model/python/BUILD.bazel

@@ -23,6 +23,9 @@ gematria_py_test(
     name = "inference_test",
     size = "small",
     srcs = ["inference_test.py"],
+    tags = [
+        "manual",
+    ],
     deps = [
         ":inference",
         ":model_base",
@@ -47,6 +50,9 @@ gematria_py_test(
     size = "small",
     timeout = "moderate",
     srcs = ["loss_utils_test.py"],
+    tags = [
+        "manual",
+    ],
     deps = [
         ":loss_utils",
         ":options",
@@ -78,6 +84,9 @@ gematria_py_test(
     size = "small",
     timeout = "moderate",
     srcs = ["main_function_test.py"],
+    tags = [
+        "manual",
+    ],
     deps = [
         ":inference",
         ":main_function",
@@ -112,6 +121,9 @@ gematria_py_test(
     size = "small",
     timeout = "moderate",
     srcs = ["model_base_test.py"],
+    tags = [
+        "manual",
+    ],
     deps = [
         ":model_base",
         ":options",
@@ -134,6 +146,9 @@ gematria_py_test(
     size = "small",
     timeout = "moderate",
     srcs = ["model_blocks_test.py"],
+    tags = [
+        "manual",
+    ],
     deps = [
         ":model_blocks",
     ],
@@ -179,6 +194,9 @@ gematria_py_test(
     size = "small",
     timeout = "moderate",
     srcs = ["token_model_test.py"],
+    tags = [
+        "manual",
+    ],
     deps = [
         ":oov_token_behavior",
         ":token_model",
@@ -224,6 +242,9 @@ gematria_py_test(
     name = "training_test",
     size = "small",
     srcs = ["training_test.py"],
+    tags = [
+        "manual",
+    ],
     deps = [
         ":training",
         "//gematria/testing/python:basic_blocks_with_throughput",

gematria/sequence/python/BUILD.bazel

@@ -38,6 +38,9 @@ gematria_py_test(
     timeout = "moderate",
     srcs = ["sequence_model_test.py"],
     shard_count = 10,
+    tags = [
+        "manual",
+    ],
     deps = [
         ":sequence_model",
         "//gematria/basic_block/python:basic_block",
@@ -66,6 +69,9 @@ gematria_py_test(
     timeout = "moderate",
     srcs = ["sequence_model_hlstm_test.py"],
     shard_count = 15,
+    tags = [
+        "manual",
+    ],
     deps = [
         ":sequence_model_hlstm",
         "//gematria/model/python:oov_token_behavior",