Skip to content

Commit 4136c57

Browse files
author
The TensorFlow Datasets Authors
committed
FeaturesDict in CroissantBuilder should use fields ids as keys, instead of names.
PiperOrigin-RevId: 680959225
1 parent d19275b commit 4136c57

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow_datasets/core/dataset_builders/croissant_builder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -244,7 +244,7 @@ def get_features(self) -> Optional[feature_lib.FeatureConnector]:
244244
)
245245
if field.repeated:
246246
feature = sequence_feature.Sequence(feature)
247-
features[field.name] = feature
247+
features[field.id] = feature
248248
return features_dict.FeaturesDict(features)
249249

250250
def _split_generators(

0 commit comments

Comments
 (0)