fix unit test

facebookresearch · Mar 6, 2021 · 767ad3d · 767ad3d
1 parent d8c1397
commit 767ad3d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/parlai/core/torch_agent.py b/parlai/core/torch_agent.py
@@ -1357,7 +1357,9 @@ def _vectorize_text(
             Truncate from the left side (keep the rightmost tokens). You
             probably want this True for inputs, False for targets.
         """
-        return self._vectorize_text_with_truncate_stats(**locals())[0]
+        return self._vectorize_text_with_truncate_stats(
+            text, add_start, add_end, truncate, truncate_left
+        )[0]
 
     def _check_truncate(self, vec, truncate, truncate_left=False):
         """