Formatted changes

patrick-toulme · Dec 17, 2024 · 1e77e67 · 1e77e67
1 parent 6f5ae2d
commit 1e77e67
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/axlearn/common/input_lm.py b/axlearn/common/input_lm.py
@@ -1178,4 +1178,4 @@ def pad_prefix():
         # 2. Control max decode length by prefix.shape[-1].
         config_for_function(pad_prefix),
         is_training=False,
-    )
+    )
diff --git a/axlearn/common/input_tf_data.py b/axlearn/common/input_tf_data.py
@@ -1471,4 +1471,4 @@ def trim_and_pad_tensor(
         t = tf.pad(t, [(0, 0)] * (len(t.shape) - 1) + [(0, pad_amt)], constant_values=pad_id)
     t = tf.ensure_shape(t, t.shape[:-1] + [max_len])
 
-    return t
+    return t