diff --git a/mala/datahandling/data_shuffler.py b/mala/datahandling/data_shuffler.py index 55074c1d..c3f71644 100644 --- a/mala/datahandling/data_shuffler.py +++ b/mala/datahandling/data_shuffler.py @@ -553,7 +553,7 @@ def shuffle_snapshots( number_of_shuffled_snapshots = self.nr_snapshots # Currently, the openPMD interface is not feature-complete. - if np.any( + if snapshot_type == "openpmd" and np.any( np.array( [ snapshot.grid_dimension[0] % number_of_shuffled_snapshots