internal

PiperOrigin-RevId: 681081771
google · Oct 3, 2024 · afbc55c · afbc55c
1 parent db0b9be
commit afbc55c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 20 deletions.
diff --git a/langfun/core/eval/matching.py b/langfun/core/eval/matching.py
@@ -183,19 +183,6 @@ def save(
           os.path.join(self.dir, Matching.MATCHES_JSON),
       )
 
-      # Save mismatches.
-      pg.save(
-          [
-              # We force the output to be dict as its type may be defined
-              # within functors which could be deserialized.
-              pg.symbolic.deref(
-                  pg.Dict(input=input, output=output), recursive=True
-              )
-              for input, output, _ in self.mismatches
-          ],
-          os.path.join(self.dir, Matching.MISMATCHES_JSON),
-      )
-
     if report:
       pg.save(
           self._html([self._render_result, self._render_matches]),

diff --git a/langfun/core/eval/matching_test.py b/langfun/core/eval/matching_test.py
@@ -157,13 +157,6 @@ def test_run(self):
             os.path.join(s.dir, matching.Matching.MATCHES_JSON)
         )
     )
-    self.assertTrue(
-        os.path.exists(
-            os.path.join(
-                s.dir, matching.Matching.MISMATCHES_JSON
-            )
-        )
-    )
     self.assertTrue(
         os.path.exists(
             os.path.join(

diff --git a/langfun/core/llms/vertexai.py b/langfun/core/llms/vertexai.py
@@ -416,6 +416,9 @@ def get_text_generation_model(
     'text/plain',
     'text/csv',
     'text/html',
+    'text/xml',
+    'text/x-script.python',
+    'application/json',
 ]