Removing tabulate dependency (#13)

valence-labs · Sep 26, 2024 · 22cb1d5 · 22cb1d5
1 parent 2b9b296
commit 22cb1d5
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/mess/basis.py b/mess/basis.py
@@ -8,7 +8,6 @@
 import pandas as pd
 from jax import jit
 from jax.ops import segment_sum
-from tabulate import tabulate
 
 from mess.orbital import Orbital, batch_orbitals
 from mess.primitive import Primitive
@@ -74,9 +73,7 @@ def __call__(self, pos: FloatNx3) -> FloatNxM:
         return orb.T
 
     def __repr__(self) -> str:
-        df = self.to_dataframe()
-        headers = ["primitive"] + df.columns.to_list()
-        return tabulate(df, headers)
+        return repr(self.to_dataframe())
 
     def _repr_html_(self) -> str | None:
         df = self.to_dataframe()