From 01c496ec2e4188c6c86a62368211fef28f2582bc Mon Sep 17 00:00:00 2001
From: Andras Schmelczer <andras@schmelczer.dev>
Date: Sat, 9 Jul 2022 21:33:31 +0200
Subject: [PATCH] Add custom repr string for better notebook DX

---
 great_ai/views/trace.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/great_ai/views/trace.py b/great_ai/views/trace.py
index 1d11bd1..2e76e20 100644
--- a/great_ai/views/trace.py
+++ b/great_ai/views/trace.py
@@ -80,3 +80,6 @@ class Trace(Generic[T], HashableBaseModel):
             "feedback_flat": self.feedback_flat,
             "tags_flat": self.tags_flat,
         }
+
+    def __repr__(self) -> str:
+        return f"Trace[{type(self.output).__name__}]({pformat(self.dict(), indent=2, compact=True)})"