allow ensembling of models without source vocabulary

neulab · May 15, 2018 · 8f807bb · 8f807bb
1 parent 421850e
commit 8f807bb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/xnmt/translator.py b/xnmt/translator.py
@@ -512,8 +512,9 @@ def __init__(self, models, src_reader, trg_reader, inference=bare(SimpleInferenc
 
     # perform checks to verify the models can logically be ensembled
     for i, model in enumerate(self.models):
-      assert self.src_reader.vocab.is_compatible(model.src_reader.vocab), \
-        f"src_reader.vocab is not compatible with model {i}"
+      if hasattr(self.src_reader, "vocab") or hasattr(model.src_reader, "vocab"):
+        assert self.src_reader.vocab.is_compatible(model.src_reader.vocab), \
+          f"src_reader.vocab is not compatible with model {i}"
       assert self.trg_reader.vocab.is_compatible(model.trg_reader.vocab), \
         f"trg_reader.vocab is not compatible with model {i}"