internal

PiperOrigin-RevId: 674890668
google · Sep 15, 2024 · e097dd5 · e097dd5
1 parent e626b0c
commit e097dd5
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/langfun/core/eval/matching.py b/langfun/core/eval/matching.py
@@ -176,7 +176,9 @@ def save(
       # Save matches.
       pg.save(
           [
-              pg.Dict(input=input, output=output)
+              pg.symbolic.deref(
+                  pg.Dict(input=input, output=output), recursive=True
+              )
               for input, output, _ in self.matches
           ],
           os.path.join(self.dir, Matching.MATCHES_JSON),
@@ -190,7 +192,9 @@ def save(
           [
               # We force the output to be dict as its type may be defined
               # within functors which could be deserialized.
-              pg.Dict(input=input, output=output)
+              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),