log shape of data being pre loaded into mem by sup3rx

NREL · Nov 4, 2024 · 8085a43 · 8085a43
1 parent ba3b226
commit 8085a43
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sup3r/preprocessing/accessor.py b/sup3r/preprocessing/accessor.py
@@ -240,7 +240,8 @@ def compute(self, **kwargs):
 
             for f in self._ds.data_vars:
                 self._ds[f] = self._ds[f].compute(**kwargs)
-                logger.debug(f'Loaded {f} into memory. {_mem_check()}')
+                logger.debug(f'Loaded {f} into memory with shape '
+                             f'{self._ds[f].shape}. {_mem_check()}')
             logger.debug(f'Loaded dataset into memory: {self._ds}')
             logger.debug(f'Post-loading: {_mem_check()}')
         return self