folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix bug (#465)

Browse Source
This commit is contained in:
Filippo Olivo
2025-03-01 12:04:25 +01:00
committed by Nicola Demo
parent 375f7f8e2d
commit d68b51fce3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pina/data/data_module.py
View File

@@ -308,7 +308,7 @@ class PinaDataModule(LightningDataModule):
if k != "equation" if k != "equation"
# Equations are NEVER dataloaded # Equations are NEVER dataloaded
} }
if offset + stage_len > len_condition: if offset + stage_len >= len_condition:
offset = len_condition - 1 offset = len_condition - 1
continue continue
offset += stage_len offset += stage_len