Better __repr__

Xeadriel · Nov 8, 2022 · 1232569 · 1232569
1 parent 194b21a
commit 1232569
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/firelang/models/_fireword.py b/firelang/models/_fireword.py
@@ -252,7 +252,7 @@ def __matmul__(self, other: FIRETensor):
 
     def __repr__(self):
         return (
-            f"<(funcs={self.funcs.__class__.__name__}, "
+            f"<FIRETensor(funcs={self.funcs.__class__.__name__}, "
             f"measures={self.measures.__class__.__name__}), "
             f"shape={self.funcs.shape}>"
         )