andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

Add custom repr string for better notebook DX

Browse source
This commit is contained in:
Andras Schmelczer 2022-07-09 21:33:31 +02:00
parent 286ede5fd3
commit 01c496ec2e
No known key found for this signature in database
GPG key ID: 0EA1BC97D0AB076E
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
great_ai/views/trace.py
View file

@ -80,3 +80,6 @@ class Trace(Generic[T], HashableBaseModel):
"feedback_flat": self.feedback_flat, "feedback_flat": self.feedback_flat,
"tags_flat": self.tags_flat, "tags_flat": self.tags_flat,
} }
def __repr__(self) -> str:
return f"Trace[{type(self.output).__name__}]({pformat(self.dict(), indent=2, compact=True)})"