Trim segement lengths to correct number of elements

qutech · Sep 29, 2023 · 2db12a3 · 2db12a3
1 parent 2a9a83b
commit 2db12a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qupulse/_program/tabor.py b/qupulse/_program/tabor.py
@@ -491,7 +491,7 @@ def _calc_sampled_segments(self) -> Tuple[Tuple[Sequence[TaborSegment], Sequence
                                    marker_b=marker_b)
             segment_idx = segments.setdefault(segment, len(segments))
             waveform_to_segment.append(segment_idx)
-        return (list(segments.keys()), segment_lengths), waveform_to_segment
+        return (list(segments.keys()), segment_lengths[:len(segments)]), waveform_to_segment
 
     def setup_single_sequence_mode(self) -> None:
         assert self.program.depth() == 1