Skip to content

Commit 7a7460a

Browse files
marcenacpThe TensorFlow Datasets Authors
authored and
The TensorFlow Datasets Authors
committed
Set both names and IDs in mlc's test fixtures.
PiperOrigin-RevId: 679138358
1 parent 9b128bf commit 7a7460a

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

tensorflow_datasets/core/utils/croissant_utils_test.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ def test_get_split_recordset():
7171
),
7272
mlc.RecordSet(
7373
id='splits',
74-
key='name',
74+
key='splits/name',
7575
data_types=[mlc.DataType.SPLIT],
7676
fields=[
7777
mlc.Field(
@@ -95,7 +95,7 @@ def test_get_split_recordset_with_no_split_recordset():
9595
record_sets = [
9696
mlc.RecordSet(
9797
id='labels',
98-
key='name',
98+
key='labels/label',
9999
fields=[
100100
mlc.Field(
101101
id='labels/label',

tensorflow_datasets/testing/test_utils.py

+4
Original file line numberDiff line numberDiff line change
@@ -736,6 +736,7 @@ def dummy_croissant_file(
736736

737737
fields = [
738738
mlc.Field(
739+
id='index',
739740
name='index',
740741
description='The sample index.',
741742
data_types=mlc.DataType.INTEGER,
@@ -745,6 +746,7 @@ def dummy_croissant_file(
745746
),
746747
),
747748
mlc.Field(
749+
id='text',
748750
name='text',
749751
description='The dummy sample text.',
750752
data_types=mlc.DataType.TEXT,
@@ -758,6 +760,7 @@ def dummy_croissant_file(
758760
record_sets = [
759761
mlc.RecordSet(
760762
id='jsonl',
763+
name='jsonl',
761764
description='Dummy record set.',
762765
fields=fields,
763766
)
@@ -778,6 +781,7 @@ def dummy_croissant_file(
778781
distribution = [
779782
mlc.FileObject(
780783
id='raw_data',
784+
name='raw_data',
781785
description='File with the data.',
782786
encoding_format='application/jsonlines',
783787
content_url=f'data/{raw_data_filename}',

0 commit comments

Comments
 (0)