From d68b51fce340fb342ba9ab4ab3f7163e6c90e0eb Mon Sep 17 00:00:00 2001
From: Filippo Olivo <filippo@filippoolivo.com>
Date: Sat, 1 Mar 2025 12:04:25 +0100
Subject: [PATCH] Fix bug (#465)

---
 pina/data/data_module.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pina/data/data_module.py b/pina/data/data_module.py
index ef6a6de..288e00e 100644
--- a/pina/data/data_module.py
+++ b/pina/data/data_module.py
@@ -308,7 +308,7 @@ class PinaDataModule(LightningDataModule):
                 if k != "equation"
                 # Equations are NEVER dataloaded
             }
-            if offset + stage_len > len_condition:
+            if offset + stage_len >= len_condition:
                 offset = len_condition - 1
                 continue
             offset += stage_len