more tests and refactoring mixin classes

NREL · Aug 17, 2023 · c67e249 · c67e249
1 parent 6bf57be
commit c67e249
Show file tree

Hide file tree

Showing 9 changed files with 2,120 additions and 1,383 deletions.
diff --git a/sup3r/pipeline/forward_pass.py b/sup3r/pipeline/forward_pass.py
diff --git a/sup3r/preprocessing/batch_handling.py b/sup3r/preprocessing/batch_handling.py
@@ -1394,9 +1394,9 @@ def __next__(self):
             )
             val_indices = self.val_indices[self._i]
             for i, idx in enumerate(val_indices):
-                high_res[i, ...] = self.handlers[idx['handler_index']].data[
-                    idx['tuple_index']
-                ][..., 0, :]
+                high_res[i, ...] = self.data_handlers[
+                    idx['handler_index']
+                ].data[idx['tuple_index']][..., 0, :]
 
             batch = self.BATCH_CLASS.get_coarse_batch(
                 high_res,