From 154c19437b0b25a37d060a5e759e092555f1d345 Mon Sep 17 00:00:00 2001
From: FilippoOlivo <filippo@filippoolivo.com>
Date: Wed, 26 Feb 2025 17:06:54 +0100
Subject: [PATCH] Fix bug

---
 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 ef6a6deb3..288e00e2e 100644
--- a/pina/data/data_module.py
+++ b/pina/data/data_module.py
@@ -308,7 +308,7 @@ def _split_condition(condition_dict, splits_dict):
                 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