From 0dea771a8c3759f2dc6636b3f0c1bf0bc07ba648 Mon Sep 17 00:00:00 2001
From: hanamoto <hanamoto@leapmind.io>
Date: Mon, 27 Apr 2020 23:06:50 +0900
Subject: [PATCH] fix

---
 src/dataset_test.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/dataset_test.py b/src/dataset_test.py
index 3256708..686d963 100644
--- 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)