``` class Dataset(object): def __init__(self):' (train_images, train_labels), (test_images, test_labels) = keras.datasets.fashion_mnist.load_data() self.test_images = test_images.astype(np.float32) / 255.0 self.labels = test_labels pass def __getitem__(self, index): return self.test_images[index], self.labels[index] def __len__(self): return len(self.test_images) ``` this code will return test dataset only