Skip to content

Commit 4ce363a

Browse files
tomvdwThe TensorFlow Datasets Authors
authored andcommitted
Fix handling map proto fields
PiperOrigin-RevId: 911386798
1 parent a07ea84 commit 4ce363a

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

tensorflow_datasets/core/dataset_info.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -790,11 +790,11 @@ def read_from_directory(self, dataset_info_dir: epath.PathLike) -> None:
790790
"is_repeated",
791791
getattr(field, "label", None) == getattr(field, "LABEL_REPEATED", 3),
792792
)
793-
if is_repeated:
794-
del field_value[:]
795-
field_value.extend(field_value_restored)
796-
elif field.type == field.TYPE_MESSAGE:
793+
if field.type == field.TYPE_MESSAGE:
797794
field_value.MergeFrom(field_value_restored)
795+
elif is_repeated:
796+
field_value[:] = []
797+
field_value.extend(field_value_restored)
798798
else:
799799
setattr(self._info_proto, field_name, field_value_restored)
800800

0 commit comments

Comments
 (0)