typos

Author: kylebgorman

Diff for: yoyodyne/models/expert.py

@@ -461,10 +461,10 @@ def _generate_data(
         dataset: data.Dataset,
         index: data.Index,
     ) -> Iterator[Tuple[List[int], List[int]]]:
-        """Helper function to manage data encoding for SED."
+        """Helper function to manage data encoding for SED.
 
-        We want encodings without BOS or EOS tokens. This encodes only raw
-        source-target text for the Maxwell library.
+        We want encodings without padding. This encodes only raw source-target
+        text for the Maxwell library.
 
         Args:
             dataset (data.Dataset): dataset for generating expert vocabulary.

Diff for: yoyodyne/predict.py

@@ -103,7 +103,7 @@ def predict(
             # Beam search.
             tsv_writer = csv.writer(sink, delimiter="\t")
             for predictions, scores in trainer.predict(model, loader):
-                predictions = util.pad_tensor_after_eos(predictions)
+                predictions = util.pad_tensor_after_end(predictions)
                 # TODO: beam search requires singleton batches and this
                 # assumes that. Revise if that restriction is ever lifted.
                 targets = [