patch

eole/inputters/text_corpus.py

@@ -140,7 +140,7 @@ def __str__(self):
 class ParallelCorpus(object):
     """A parallel corpus file pair that can be loaded to iterate."""
 
-    def __init__(self, name, src, tgt, sco=None, align=None):
+    def __init__(self, name, src, tgt, sco=None, align=None, is_train=False):
         """Initialize src & tgt side file path."""
         self.id = name
         self.src = src
@@ -314,7 +314,7 @@ class ParallelCorpusIterator(object):
         offset (int): iterate corpus with this line offset.
     """
 
-    def __init__(self, corpus, transform, skip_empty_level="warning", stride=1, offset=0):
+    def __init__(self, corpus, transform, skip_empty_level="warning", stride=1, offset=0, is_train=False):
         self.cid = corpus.id
         self.corpus = corpus
         self.transform = transform