typos

CUNY-CL · Dec 2, 2024 · 8bc0127 · 8bc0127
1 parent 3543096
commit 8bc0127
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/yoyodyne/models/expert.py b/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 --git a/yoyodyne/predict.py b/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 = [