Skip to content

Commit 358fc6d

Browse files
authored
check that prototype datasets are serializable (#5557)
1 parent b2d42a2 commit 358fc6d

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

test/test_prototype_builtin_datasets.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import functools
22
import io
3+
import pickle
34
from pathlib import Path
45

56
import pytest
@@ -108,12 +109,12 @@ def test_transformable(self, test_home, dataset_mock, config):
108109
next(iter(dataset.map(transforms.Identity())))
109110

110111
@parametrize_dataset_mocks(DATASET_MOCKS)
111-
def test_traversable(self, test_home, dataset_mock, config):
112+
def test_serializable(self, test_home, dataset_mock, config):
112113
dataset_mock.prepare(test_home, config)
113114

114115
dataset = datasets.load(dataset_mock.name, **config)
115116

116-
traverse(dataset)
117+
pickle.dumps(dataset)
117118

118119
@parametrize_dataset_mocks(DATASET_MOCKS)
119120
@pytest.mark.parametrize("annotation_dp_type", (Shuffler, ShardingFilter))

0 commit comments

Comments
 (0)