diff --git a/CHANGELOG.md b/CHANGELOG.md
index d976ad4b9..6e3c06d8d 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -57,6 +57,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Maintenance
 
+- Fix reST formatting errors in `ARModel` methods to ensure documentation builds correctly. [\#458](https://github.com/mllam/neural-lam/pull/458) @sohampatil01-svg
+
 - Update PR template to clarify milestone/roadmap requirement and maintenance changes [\#186](https://github.com/mllam/neural-lam/pull/186) @joeloskarsson
 
 - Update CI/CD to use python 3.13 for testing and full range of current python versions for linting (3.10 - 3.14) [\#173](https://github.com/mllam/neural-lam/pull/173) @observingClouds
diff --git a/neural_lam/models/ar_model.py b/neural_lam/models/ar_model.py
index f1bcb461d..aeed3720f 100644
--- a/neural_lam/models/ar_model.py
+++ b/neural_lam/models/ar_model.py
@@ -278,11 +278,14 @@ def unroll_prediction(self, init_states, forcing_features, true_states):
 
     def common_step(self, batch):
         """
-        Predict on single batch batch consists of: init_states: (B, 2,
-        num_grid_nodes, d_features) target_states: (B, pred_steps,
-        num_grid_nodes, d_features) forcing_features: (B, pred_steps,
-        num_grid_nodes, d_forcing),
-            where index 0 corresponds to index 1 of init_states
+        Predict on single batch.
+
+        Batch consists of:
+            init_states: (B, 2, num_grid_nodes, d_features)
+            target_states: (B, pred_steps, num_grid_nodes, d_features)
+            forcing_features: (B, pred_steps, num_grid_nodes, d_forcing)
+
+        Note: index 0 corresponds to index 1 of init_states.
         """
         (init_states, target_states, forcing_features, batch_times) = batch