Skip to content

Commit da59d68

Browse files
authored
better features repr (#7640)
1 parent 5b1d12d commit da59d68

File tree

1 file changed

+18
-3
lines changed

1 file changed

+18
-3
lines changed

src/datasets/features/features.py

Lines changed: 18 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -476,7 +476,7 @@ def cast_to_python_objects(obj: Any, only_1d_for_numpy=False, optimize_list_cast
476476
)[0]
477477

478478

479-
@dataclass
479+
@dataclass(repr=False)
480480
class Value:
481481
"""
482482
Scalar feature value of a particular data type.
@@ -551,6 +551,9 @@ def encode_example(self, value):
551551
else:
552552
return value
553553

554+
def __repr__(self):
555+
return f"{type(self).__name__}('{self.dtype}')"
556+
554557

555558
class _ArrayXD:
556559
def __post_init__(self):
@@ -1182,7 +1185,7 @@ def Sequence(feature, length=-1):
11821185
return List(feature, length=length)
11831186

11841187

1185-
@dataclass
1188+
@dataclass(repr=False)
11861189
class List:
11871190
"""Feature type for large list data composed of child feature data type.
11881191
@@ -1203,8 +1206,14 @@ class List:
12031206
pa_type: ClassVar[Any] = None
12041207
_type: str = field(default="List", init=False, repr=False)
12051208

1209+
def __repr__(self):
1210+
if self.length != -1:
1211+
return f"{type(self).__name__}({self.feature}, length={self.length})"
1212+
else:
1213+
return f"{type(self).__name__}({self.feature})"
1214+
12061215

1207-
@dataclass
1216+
@dataclass(repr=False)
12081217
class LargeList:
12091218
"""Feature type for large list data composed of child feature data type.
12101219
@@ -1221,6 +1230,12 @@ class LargeList:
12211230
pa_type: ClassVar[Any] = None
12221231
_type: str = field(default="LargeList", init=False, repr=False)
12231232

1233+
def __repr__(self):
1234+
if self.length != -1:
1235+
return f"{type(self).__name__}({self.feature}, length={self.length})"
1236+
else:
1237+
return f"{type(self).__name__}({self.feature})"
1238+
12241239

12251240
FeatureType = Union[
12261241
dict,

0 commit comments

Comments
 (0)