Port model_base to TF2

This patch ports most of the functionality of model_base to TF2 along with all the tests. Notably left out is run_continuous_evaluation as it is tested separately. This will be fixed in a future patch where the tests for that are updated as well. Pull Request: google#280
boomanaiden154 · Jan 4, 2025 · 8b637ff · 8b637ff
1 parent c7639d1
commit 8b637ff
Show file tree

Hide file tree

Showing 4 changed files with 363 additions and 677 deletions.
diff --git a/gematria/model/python/BUILD.bazel b/gematria/model/python/BUILD.bazel
@@ -118,9 +118,6 @@ gematria_py_test(
     size = "small",
     timeout = "moderate",
     srcs = ["model_base_test.py"],
-    tags = [
-        "manual",
-    ],
     deps = [
         ":model_base",
         ":options",