Fix traces table filtering for string arguments

2022-08-11 16:49:19 +02:00 · 2022-08-11 16:49:19 +02:00 · 49968691cc
commit 49968691cc
parent 83e111f088
2 changed files with 10 additions and 3 deletions
--- a/great_ai/views/trace.py
+++ b/great_ai/views/trace.py
@ -88,7 +88,9 @@ class Trace(Generic[T], HashableBaseModel):
                }
            ),
            **{
-                k: pformat(v, indent=2, compact=True)
+                k: v
+                if (isinstance(v, float) or isinstance(v, int))
+                else pformat(v, indent=2, compact=True)
                for k, v in self.logged_values.items()
            },
            "models_flat": self.models_flat,