Skip to content

GitLab

Commit 152c2b3a authored by Vishwak Srinivasan's avatar Vishwak Srinivasan Committed by Soumith Chintala
Browse files

Fix lint error (#588)

parent d9e50b17
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
test/test_folder.py
View file @ 152c2b3a
...@@ -16,8 +16,10 @@ def mock_transform(return_value, arg_list): ...@@ -16,8 +16,10 @@ def mock_transform(return_value, arg_list):
class Tester(unittest.TestCase): class Tester(unittest.TestCase):
root = get_file_path_2('test/assets/dataset/') root = get_file_path_2('test/assets/dataset/')
classes = ['a', 'b'] classes = ['a', 'b']
class_a_images = [get_file_path_2(os.path.join('test/assets/dataset/a/', path)) for path in ['a1.png', 'a2.png', 'a3.png']] class_a_images = [get_file_path_2(os.path.join('test/assets/dataset/a/', path))
class_b_images = [get_file_path_2(os.path.join('test/assets/dataset/b/', path)) for path in ['b1.png', 'b2.png', 'b3.png', 'b4.png']] for path in ['a1.png', 'a2.png', 'a3.png']]
class_b_images = [get_file_path_2(os.path.join('test/assets/dataset/b/', path))
for path in ['b1.png', 'b2.png', 'b3.png', 'b4.png']]
def test_image_folder(self): def test_image_folder(self):
dataset = ImageFolder(Tester.root, loader=lambda x: x) dataset = ImageFolder(Tester.root, loader=lambda x: x)
......
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