Defining _DMMSchedule

pasqal-io · Jul 24, 2023 · 789080f · 789080f
1 parent 4ca5a20
commit 789080f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pulser-core/pulser/sequence/_schedule.py b/pulser-core/pulser/sequence/_schedule.py
@@ -24,6 +24,7 @@
 from pulser.channels.base_channel import Channel
 from pulser.pulse import Pulse
 from pulser.register.base_register import QubitId
+from pulser.register.weight_maps import DetuningMap
 from pulser.sampler.samples import ChannelSamples, _PulseTargetSlot
 from pulser.waveforms import ConstantWaveform
 
@@ -247,6 +248,11 @@ def __iter__(self) -> Iterator[_TimeSlot]:
             yield slot
 
 
+@dataclass
+class _DMMSchedule(_ChannelSchedule):
+    detuning_map: DetuningMap
+
+
 class _Schedule(Dict[str, _ChannelSchedule]):
     def __init__(self, max_duration: int | None = None):
         self.max_duration = max_duration