add missing import

Author: pmeier

test/test_datasets.py

@@ -17,6 +17,7 @@
 import itertools
 import datasets_utils
 import pathlib
+import pickle
 from torchvision import datasets