def unpickle(file):

fo = open(file, 'rb')
dict = pickle.load(fo, encoding ='bytes')
X = dict[b'data']
fo.close()
return dict

unpickle('datasets/cifar-10-batches-py/test_batch')