Skip to content

Commit 223f180

Browse files
committed
Fixed issue with table representation for empty Datasets
1 parent fc56b98 commit 223f180

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

src/omnipy/data/dataset.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -595,12 +595,19 @@ def __repr__(self):
595595
def _trad_repr(self) -> str:
596596
return super().__repr__()
597597

598+
@classmethod
599+
def _len_if_available(cls, obj: Any) -> int | str:
600+
try:
601+
return len(obj)
602+
except TypeError:
603+
return 'N/A'
604+
598605
def _table_repr(self) -> str:
599606
ret = tabulate(
600607
((i,
601608
k,
602609
type(v).__name__,
603-
len(v) if hasattr(v, '__len__') else 'N/A',
610+
self._len_if_available(v),
604611
humanize.naturalsize(objsize.get_deep_size(v)))
605612
for i, (k, v) in enumerate(self.items())),
606613
('#', 'Data file name', 'Type', 'Length', 'Size (in memory)'),

0 commit comments

Comments
 (0)