Port loss_utils_test to TF2

The loss_utils class does not need any changes to support TF2, so this patch simply migrates all the test cases to use eager execution. This patch does not make the class idiomatically TF2 however. That will be a later patch once everything else has been ported to allow for incrementally improving the code while validating with all the tests. Pull Request: google#279
boomanaiden154 · Jan 3, 2025 · c7639d1 · c7639d1
1 parent 88d5960
commit c7639d1
Show file tree

Hide file tree

Showing 2 changed files with 172 additions and 250 deletions.
diff --git a/gematria/model/python/BUILD.bazel b/gematria/model/python/BUILD.bazel
@@ -50,9 +50,6 @@ gematria_py_test(
     size = "small",
     timeout = "moderate",
     srcs = ["loss_utils_test.py"],
-    tags = [
-        "manual",
-    ],
     deps = [
         ":loss_utils",
         ":options",