Merge branch 'main' into modes/mobilenetv2_weights

pytorch · Mar 7, 2022 · bb14c86 · bb14c86
2 parents 43657d1 + 358fc6d
commit bb14c86
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/test_prototype_builtin_datasets.py b/test/test_prototype_builtin_datasets.py
@@ -1,5 +1,6 @@
 import functools
 import io
+import pickle
 from pathlib import Path
 
 import pytest
@@ -108,12 +109,12 @@ def test_transformable(self, test_home, dataset_mock, config):
         next(iter(dataset.map(transforms.Identity())))
 
     @parametrize_dataset_mocks(DATASET_MOCKS)
-    def test_traversable(self, test_home, dataset_mock, config):
+    def test_serializable(self, test_home, dataset_mock, config):
         dataset_mock.prepare(test_home, config)
 
         dataset = datasets.load(dataset_mock.name, **config)
 
-        traverse(dataset)
+        pickle.dumps(dataset)
 
     @parametrize_dataset_mocks(DATASET_MOCKS)
     @pytest.mark.parametrize("annotation_dp_type", (Shuffler, ShardingFilter))