From ef00869f23c065bc135c1d244b32004f7c593722 Mon Sep 17 00:00:00 2001 From: Soham_patil Date: Fri, 20 Mar 2026 11:43:42 +0530 Subject: [PATCH 1/2] docs: Fix docstring formatting in ARModel --- neural_lam/models/ar_model.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) 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 From d2c27ef08baa9509ec916141e74f16ad6cb16a80 Mon Sep 17 00:00:00 2001 From: Soham_patil Date: Fri, 20 Mar 2026 12:49:01 +0530 Subject: [PATCH 2/2] docs: Update CHANGELOG.md for PR #458 --- CHANGELOG.md | 2 ++ 1 file changed, 2 insertions(+) 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