fix decode in encoder decoder (#586)

allenai · Dec 7, 2017 · ccc4af3 · ccc4af3
1 parent d978101
commit ccc4af3
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/allennlp/models/encoder_decoders/simple_seq2seq.py b/allennlp/models/encoder_decoders/simple_seq2seq.py
@@ -280,8 +280,6 @@ def decode(self, output_dict: Dict[str, torch.Tensor]) -> Dict[str, torch.Tensor
             predicted_tokens = [self.vocab.get_token_from_index(x, namespace="target_tokens")
                                 for x in indices]
             all_predicted_tokens.append(predicted_tokens)
-        if len(all_predicted_tokens) == 1:
-            all_predicted_tokens = all_predicted_tokens[0]  # type: ignore
         output_dict["predicted_tokens"] = all_predicted_tokens
         return output_dict