Forgot to initialize shared in parent class of RepeatTeacher which in…

…troduced bug that we were seeing when batch size > 0
MannyKayy · Dec 21, 2017 · 5fd8c1c · 5fd8c1c
1 parent 50001aa
commit 5fd8c1c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/tasks/repeat.py b/tests/tasks/repeat.py
@@ -16,7 +16,7 @@ def __init__(self, opt, shared=None):
         opt['datafile'] = 'unused_path'
         task = opt.get('task', 'tests.tasks.repeat:RepeatTeacher:50')
         self.data_length = int(task.split(':')[2])
-        super().__init__(opt)
+        super().__init__(opt, shared)
 
     def setup_data(self, unused_path):
         for i in range(self.data_length):