fixed omission of end_indices in decode_batch return (plus deleted comment)

bsh98 · bsh98 · commit b2c3e9058494 · 2021-06-28T22:45:35.000-07:00
diff --git a/model_generator.py b/model_generator.py
@@ -132,7 +132,6 @@ def decode_batch(self, bodies, special_append=None, max_output_length=100, sampl
         elif return_scores:
             return outputs, scores.tolist()
         else:
-            # return outputs
             return outputs, end_indices
 
     def decode_beam_batch(self, bodies, beam_size=3, max_output_length=100, sample=False):
