From 327d1aca5b359f237392dbc4d9f17f83702a4eee Mon Sep 17 00:00:00 2001
From: Nicola Demo <demo.nicola@gmail.com>
Date: Wed, 29 Nov 2023 16:24:49 +0100
Subject: [PATCH] add lenght attribute for pina dataloader

---
 pina/dataset.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/pina/dataset.py b/pina/dataset.py
index 1cf2f96..38b2200 100644
--- a/pina/dataset.py
+++ b/pina/dataset.py
@@ -238,3 +238,12 @@ class SamplePointLoader:
                     'condition': self.batch_data_conditions[idx_],
                 }
             yield d
+
+    def __len__(self):
+        """
+        Return the number of batches.
+
+        :return: The number of batches.
+        :rtype: int
+        """
+        return len(self.batch_list)
\ No newline at end of file