Skip to content

Commit d561031

Browse files
committed
better __repr__ for AxisCollection with no axes
1 parent a3b2242 commit d561031

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

larray/core/axis.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1811,8 +1811,12 @@ def __str__(self):
18111811
return "{%s}" % ', '.join(self.display_names)
18121812

18131813
def __repr__(self):
1814-
axes_repr = (repr(axis) for axis in self._list)
1815-
return "AxisCollection([\n %s\n])" % ',\n '.join(axes_repr)
1814+
if len(self):
1815+
repr_per_axis = [repr(axis) for axis in self._list]
1816+
axes_repr = "\n {}\n".format(',\n '.join(repr_per_axis))
1817+
else:
1818+
axes_repr = ""
1819+
return "AxisCollection([{}])".format(axes_repr)
18161820

18171821
# TODO: kill name argument (does not seem to be used anywhere
18181822
def get(self, key, default=None, name=None):

0 commit comments

Comments
 (0)