improve performance of read_object with idx by reading in whole objec…

…t, then slicing
legend-exp · Nov 6, 2023 · ec55d3c · ec55d3c
1 parent 6901bf3
commit ec55d3c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/lgdo/lh5_store.py b/src/lgdo/lh5_store.py
@@ -741,7 +741,7 @@ def read_object(
                     tmp_shape = (0,) + h5f[name].shape[1:]
                     nda = np.empty(tmp_shape, h5f[name].dtype)
                 else:
-                    nda = h5f[name][source_sel]
+                    nda = h5f[name][...][source_sel]
 
             # special handling for bools
             # (c and Julia store as uint8 so cast to bool)