From 1e77e67293dda7867a3090eeff15d44fa1c1f000 Mon Sep 17 00:00:00 2001
From: dgourab
 <dgourab@samwise-odcr-p5-queue-dy-p548xlarge-20.samwisecmhubuntu2004sharedcluster-pcluster-v3-10-1-prod.pcluster>
Date: Tue, 17 Dec 2024 00:07:42 +0000
Subject: [PATCH] Formatted changes

---
 axlearn/common/input_lm.py      | 2 +-
 axlearn/common/input_tf_data.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/axlearn/common/input_lm.py b/axlearn/common/input_lm.py
index c93f5ed0..4eb6a881 100644
--- 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,
-    )
\ No newline at end of file
+    )
diff --git a/axlearn/common/input_tf_data.py b/axlearn/common/input_tf_data.py
index 4fc8b66d..1d44ad1b 100644
--- 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
\ No newline at end of file
+    return t