[fbsync] check that prototype datasets are serializable (#5557)

Reviewed By: vmoens

Differential Revision: D34878997

fbshipit-source-id: fd77f0c236a3fb3e1ee7ab77a27915ec0b785fe2

Author: datumbox

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))