From f0f106710ed816a1544f7eddade2ff05c72e0ca6 Mon Sep 17 00:00:00 2001 From: Hatem Helal Date: Thu, 26 Sep 2024 13:49:18 +0100 Subject: [PATCH] Removing tabulate dependency --- mess/basis.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/mess/basis.py b/mess/basis.py index e3a7ddf..b9b4b0b 100644 --- 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()