reuse dataframe repr_html for nicer notebook output

graphcore-research · Apr 29, 2024 · 4e3ba54 · 4e3ba54
1 parent e04457e
commit 4e3ba54
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/mess/basis.py b/mess/basis.py
@@ -77,6 +77,10 @@ def __repr__(self) -> str:
         headers = ["primitive"] + df.columns.to_list()
         return tabulate(df, headers)
 
+    def _repr_html_(self) -> str | None:
+        df = self.to_dataframe()
+        return df._repr_html_()
+
     def __hash__(self) -> int:
         return hash(self.primitives)