diff --git a/src/extra/components/dld.py b/src/extra/components/dld.py index 63ea626b..bbd0a0bb 100644 --- a/src/extra/components/dld.py +++ b/src/extra/components/dld.py @@ -250,7 +250,7 @@ def edges(self, channel_index=True, pulse_dim='pulseId'): index = self._align_pulse_index(kd, pulse_dim) data = kd.ndarray() raw_edges = [self._build_reduced_pd(data[:, i, :], index, None) - for i in range(7)] + for i in range(data.shape[1])] index = pd.MultiIndex.from_frame( pd.concat([e.index.to_frame() for e in raw_edges]))