Skip to content

Commit 74109ac

Browse files
author
The TensorFlow Datasets Authors
committed
Include uint in the parametrized tests for CroissantBuilder.
PiperOrigin-RevId: 742237760
1 parent 762c979 commit 74109ac

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tensorflow_datasets/core/dataset_builders/croissant_builder_test.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -100,6 +100,14 @@
100100
None,
101101
np.float64,
102102
),
103+
(
104+
mlc.Field(
105+
data_types=mlc.DataType.UINT16, description="Uint16 feature"
106+
),
107+
np.uint16,
108+
None,
109+
np.uint16,
110+
),
103111
(
104112
mlc.Field(
105113
data_types=mlc.DataType.BOOL, description="Boolean feature"

0 commit comments

Comments
 (0)