update ct dataset

nnaakkaaii · Jun 16, 2024 · 75079f8 · 75079f8
1 parent 93b433e
commit 75079f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hrdae/dataloaders/datasets/ct.py b/hrdae/dataloaders/datasets/ct.py
@@ -88,7 +88,7 @@ def __init__(
         self.in_memory = in_memory
 
     def __len__(self) -> int:
-        return len(self.data)
+        return len(self.paths)
 
     def __getitem__(self, index: int) -> dict[str, Tensor]:
         if len(self.data) > 0: