roll back tests for model forward

Author: francoishernandez

onmt/tests/test_models.py

@@ -134,7 +134,7 @@ def nmtmodel_forward(self, opt, source_l=3, bsize=1):
 
         test_src, test_tgt, test_length = self.get_batch(source_l=source_l,
                                                          bsize=bsize)
-        outputs, attn, _, _ = model(test_src, test_tgt, test_length)
+        outputs, attn = model(test_src, test_tgt, test_length)
         outputsize = torch.zeros(source_l - 1, bsize, opt.dec_rnn_size)
         # Make sure that output has the correct size and type
         self.assertEqual(outputs.size(), outputsize.size())
@@ -168,7 +168,7 @@ def imagemodel_forward(self, opt, tgt_l=2, bsize=1, h=15, w=17):
             h=h, w=w,
             bsize=bsize,
             tgt_l=tgt_l)
-        outputs, attn, _, _ = model(test_src, test_tgt, test_length)
+        outputs, attn = model(test_src, test_tgt, test_length)
         outputsize = torch.zeros(tgt_l - 1, bsize, opt.dec_rnn_size)
         # Make sure that output has the correct size and type
         self.assertEqual(outputs.size(), outputsize.size())
@@ -206,7 +206,7 @@ def audiomodel_forward(self, opt, tgt_l=7, bsize=3, t=37):
             sample_rate=opt.sample_rate,
             window_size=opt.window_size,
             t=t, tgt_l=tgt_l)
-        outputs, attn, _, _ = model(test_src, test_tgt, test_length)
+        outputs, attn = model(test_src, test_tgt, test_length)
         outputsize = torch.zeros(tgt_l - 1, bsize, opt.dec_rnn_size)
         # Make sure that output has the correct size and type
         self.assertEqual(outputs.size(), outputsize.size())