From 85db905ee4c3179b0d87b96b60ceac61abe8c15b Mon Sep 17 00:00:00 2001 From: Stephen Roller Date: Wed, 17 Oct 2018 10:19:44 -0700 Subject: [PATCH] Make sure RepeatTeacher doesn't leak files. --- tests/tasks/repeat.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/tasks/repeat.py b/tests/tasks/repeat.py index 94de4ef5fbc..508e16d058f 100644 --- a/tests/tasks/repeat.py +++ b/tests/tasks/repeat.py @@ -24,3 +24,9 @@ def __init__(self, opt, shared=None): def setup_data(self, unused_path): for i in range(self.data_length): yield ((str(i), [str(i)]), True) + + def num_examples(self): + return self.data_length + + def num_episodes(self): + return self.data_length