Improve conditions and refactor dataset classes (#475)

* Reimplement conditions * Refactor datasets and implement LabelBatch --------- Co-authored-by: Dario Coscia <dariocos99@gmail.com>
2025-03-07 11:24:09 +01:00
parent bdad144461
commit a0cbf1c44a
40 changed files with 943 additions and 550 deletions
--- a/pina/solver/solver.py
+++ b/pina/solver/solver.py
@@ -172,7 +172,7 @@ class SolverInterface(lightning.pytorch.LightningModule, metaclass=ABCMeta):
        # assuming batch is a custom Batch object
        batch_size = 0
        for data in batch:
-            batch_size += len(data[1]["input_points"])
+            batch_size += len(data[1]["input"])
        return batch_size

    @staticmethod