Commit 77eb7ea7 authored by Mikhail Korobov's avatar Mikhail Korobov Committed by Francisco Massa
Browse files

make flake8 happy (#228)

parent 3ccc1dfd
......@@ -52,7 +52,8 @@ class MNIST(data.Dataset):
self.train_data, self.train_labels = torch.load(
os.path.join(self.root, self.processed_folder, self.training_file))
else:
self.test_data, self.test_labels = torch.load(os.path.join(self.root, self.processed_folder, self.test_file))
self.test_data, self.test_labels = torch.load(
os.path.join(self.root, self.processed_folder, self.test_file))
def __getitem__(self, index):
"""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment