2
2
import pytest
3
3
4
4
import h5py
5
- from datasets import Array2D , Array3D , Array4D , Features , Sequence , Value
5
+ from datasets import Array2D , Array3D , Array4D , Features , List , Value
6
6
from datasets .builder import InvalidConfigName
7
7
from datasets .data_files import DataFilesDict , DataFilesList
8
8
from datasets .download .streaming_download_manager import StreamingDownloadManager
@@ -579,8 +579,8 @@ def test_hdf5_feature_inference(hdf5_file_with_arrays):
579
579
# (n_rows, 2, 3, 4, 5) -> Array4D with shape (2, 3, 4, 5)
580
580
assert isinstance (features ["tensor_4d" ], Array4D )
581
581
assert features ["tensor_4d" ].shape == (2 , 3 , 4 , 5 )
582
- # (n_rows, 10) -> Sequence of length 10
583
- assert isinstance (features ["vector_1d" ], Sequence )
582
+ # (n_rows, 10) -> List of length 10
583
+ assert isinstance (features ["vector_1d" ], List )
584
584
assert features ["vector_1d" ].length == 10
585
585
586
586
@@ -600,9 +600,9 @@ def test_hdf5_vlen_feature_inference(hdf5_file_with_vlen_arrays):
600
600
601
601
# Check specific feature types for variable-length arrays
602
602
features = hdf5 .info .features
603
- # Variable-length arrays should become Sequence features by default (for small datasets)
604
- assert isinstance (features ["vlen_ints" ], Sequence )
605
- assert isinstance (features ["mixed_data" ], Sequence )
603
+ # Variable-length arrays should become List features by default (for small datasets)
604
+ assert isinstance (features ["vlen_ints" ], List )
605
+ assert isinstance (features ["mixed_data" ], List )
606
606
607
607
# Check that the inner feature types are correct
608
608
assert isinstance (features ["vlen_ints" ].feature , Value )
0 commit comments