Merge branch 'fix/table' into 'release/v0.33.0'

xsuite · Apr 21, 2023 · b1b067b · b1b067b
2 parents 066a39c + b31ae00
commit b1b067b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/xtrack/table.py b/xtrack/table.py
@@ -298,6 +298,9 @@ def values(self):
     def items(self):
         return self._data.items()
 
+    def __dir__(self):
+        return super().__dir__() + list(self._data.keys())
+
     def __iter__(self):
         return self._data.__iter__()
 
@@ -335,7 +338,7 @@ def __setattr__(self, key, val):
 
     def __repr__(self):
         n = self._nrows
-        c = len(self._data)
+        c = len(self._col_names)
         ns = "s" if n != 1 else ""
         cs = "s" if c != 1 else ""
         out = [f"{self.__class__.__name__}: {n} row{ns}, {c} col{cs}"]