[𝘀𝗽𝗿] initial version

Created using spr 1.3.4
google · Jan 3, 2025 · b191206 · b191206
1 parent 09a6cde
commit b191206
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 0 deletions.
diff --git a/gematria/granite/python/BUILD.bazel b/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",

diff --git a/gematria/model/python/BUILD.bazel b/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",

diff --git a/gematria/sequence/python/BUILD.bazel b/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",