fix

LeapMind · Apr 27, 2020 · 0dea771 · 0dea771
1 parent eb25e29
commit 0dea771
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/dataset_test.py b/src/dataset_test.py
@@ -29,6 +29,6 @@ def test_dataset(self):
     def test_dataloader(self):
         self.assertEqual(len(self.dataloader_test), 2)
         ret = 0
-        for _ in self.dataloader:
+        for _ in self.dataloader_test:
             ret += 1
         self.assertEqual(ret, 2 // 2)