From b191206cc56801a6c2e30e71f53c3e828ea507c1 Mon Sep 17 00:00:00 2001 From: Aiden Grossman Date: Fri, 3 Jan 2025 21:04:35 +0000 Subject: [PATCH] =?UTF-8?q?[=F0=9D=98=80=F0=9D=97=BD=F0=9D=97=BF]=20initia?= =?UTF-8?q?l=20version?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Created using spr 1.3.4 --- gematria/granite/python/BUILD.bazel | 12 ++++++++++++ gematria/model/python/BUILD.bazel | 21 +++++++++++++++++++++ gematria/sequence/python/BUILD.bazel | 6 ++++++ 3 files changed, 39 insertions(+) diff --git a/gematria/granite/python/BUILD.bazel b/gematria/granite/python/BUILD.bazel index ec928b24..c89d87e8 100644 --- 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 index 6fe00310..bbf44300 100644 --- 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 index 6893c8d4..ae16210e 100644 --- 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",