iqm-pulse 9.19.0__tar.gz → 9.21.0__tar.gz

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/CHANGELOG.rst

@@ -2,6 +2,22 @@
 Changelog
 =========
 
+Version 9.21.0 (2025-07-10)
+===========================
+
+Bug fixes
+---------
+
+- Fix instructions with same field names being treated as equal in building the playlist
+
+Version 9.20.0 (2025-07-09)
+===========================
+
+Features
+--------
+
+- Enable mypy type checking in CI and add temporary type ignores to the source code. :issue:`SW-1615`
+
 Version 9.19.0 (2025-07-08)
 ===========================
 

{iqm_pulse-9.19.0/src/iqm_pulse.egg-info → iqm_pulse-9.21.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: iqm-pulse
-Version: 9.19.0
+Version: 9.21.0
 Summary: A Python-based project for providing interface and implementations for control pulses.
 Author-email: IQM Finland Oy <info@meetiqm.com>
 License:                                  Apache License
@@ -219,6 +219,7 @@ Requires-Dist: python-rapidjson==1.20
 Requires-Dist: jinja2==3.0.3
 Requires-Dist: numpy<3.0,>=1.26.4
 Requires-Dist: scipy<1.16,>=1.11.4
+Requires-Dist: scipy-stubs
 
 IQM Pulse library
 =================

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/pyproject.toml

@@ -21,6 +21,14 @@ file = "LICENSE.txt"
 Documentation = "https://iqm-finland.github.io/docs/iqm-pulse/"
 Homepage = "https://pypi.org/project/iqm-pulse/"
 
+[tool.mypy]
+explicit_package_bases = true
+mypy_path = "$MYPY_CONFIG_FILE_DIR/src"
+namespace_packages = true
+[[tool.mypy.overrides]]
+module = [ "iqm.data_definitions.*", "iqm.models.*", "daemon.*", "quimb.*", "qiskit.*", "qiskit_aer.*", "grpc.*", "lmfit.*", "emukit.*", "mockito.*", "sklearn.*", "dill.*", "uncertainties.*", "GPy.*", "pythonjsonlogger.*",]
+ignore_missing_imports = true
+
 [tool.ruff]
 exclude = [ "*.ipynb", "*.rst",]
 extend-include = [ "sub-setup-py.template",]

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/requirements/base.in

@@ -3,3 +3,4 @@ python-rapidjson == 1.20
 jinja2 == 3.0.3
 numpy >= 1.26.4,<3.0
 scipy >= 1.11.4,<1.16
+scipy-stubs

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/requirements/base.txt

@@ -3,4 +3,5 @@ iqm-data-definitions >= 2.13, < 3.0
 python-rapidjson == 1.20
 jinja2 == 3.0.3
 numpy >= 1.26.4,<3.0
-scipy >= 1.11.4,<1.16
+scipy >= 1.11.4,<1.16
+scipy-stubs

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/src/iqm/pulse/builder.py

@@ -407,7 +407,7 @@ class ScheduleBuilder:
 
         """
         probe_line = self.chip_topology.component_to_probe_line.get(feedback_qubit)
-        all_virtual_channels = self.get_virtual_feedback_channels(probe_line)
+        all_virtual_channels = self.get_virtual_feedback_channels(probe_line)  # type: ignore[arg-type]
         channel = next((c for c in all_virtual_channels if awg_name in c), None)
         if not channel:
             raise ValueError(f"AWG node {awg_name} does not support fast feedback from {probe_line}")
@@ -1151,7 +1151,7 @@ class ScheduleBuilder:
         components = box.neighborhood_components.get(0)
         if components is None:
             components = box.locus_components.copy()
-            for channel in box.atom:
+            for channel in box.atom:  # type: ignore[union-attr]
                 components.add(self._channel_to_component.get(channel, channel))
             box.neighborhood_components[0] = components
 
@@ -1311,7 +1311,7 @@ class ScheduleBuilder:
         }
 
         pl = Playlist()
-        mapped_instructions: dict[str, dict[int, Any]] = {}
+        mapped_instructions: dict[str, dict[int | Instruction, Any]] = {}
 
         def _append_to_schedule(sc_schedule: SC_Schedule, channel_name: str, instr: Instruction) -> None:
             """Append ``instr`` to ``sc_schedule`` into the channel``channel_name``."""
@@ -1320,10 +1320,12 @@ class ScheduleBuilder:
                 # 2 dataclasses can have the same hash if their fields are identical. We must
                 # distinguish between different Waveform classes which may have identical fields,
                 # so we use the instruction itself as a key, so that the class is checked too.
-                instr_id = hash(instr)
+                instr_id = instr  # type: ignore[attr-defined]
+                is_mapped = instr_id in mapped_instructions.setdefault(channel_name, {})
             except TypeError:
-                instr_id = instr.id
-            if instr_id not in mapped_instructions.setdefault(channel_name, {}):
+                instr_id = instr.id  # type: ignore[attr-defined]
+                is_mapped = instr_id in mapped_instructions.setdefault(channel_name, {})
+            if not is_mapped:
                 mapped = _map_instruction(instr)
                 idx = pl.channel_descriptions[channel_name].add_instruction(mapped)
                 sc_schedule.instructions.setdefault(channel_name, []).append(idx)

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/src/iqm/pulse/circuit_operations.py

@@ -278,7 +278,7 @@ class CircuitOperationList(list):
             if self.table[op_name].arity:
                 self._set_specific_operation_shortcut(op_name)
 
-    def __getitem__(self, item) -> CircuitOperationList | CircuitOperation:
+    def __getitem__(self, item) -> CircuitOperationList | CircuitOperation:  # type: ignore[override]  # type: ignore[override]  # type: ignore[override]
         """For the builtin list, this method is used both for accessing a single element: ``mylist[0]`` and accessing
         a slice: ``mylist[1:3]``. The latter should generate a new CircuitOperationList, so we override the method to
         ensure that it does.

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/src/iqm/pulse/gate_implementation.py

@@ -169,7 +169,7 @@ class GateImplementation(abc.ABC):
         Inheriting classes may override this method if the default :meth:`__call__` caching (based on the args & kwargs
         in the signature) is sufficient. Any additional caching may also be implemented inside this function if needed.
         """
-        return NotImplementedError
+        return NotImplementedError  # type: ignore[return-value]
 
     def build(
         self, op_name: str, locus: Locus, impl_name: str | None = None, strict_locus: bool = False
@@ -452,7 +452,7 @@ class CustomIQWaveforms(GateImplementation):
                 if k not in cls.excluded_parameters
             }
             if cls.dependent_waves:
-                cls.parameters = root_parameters | parameters_i
+                cls.parameters = root_parameters | parameters_i  # type: ignore[assignment]
             else:
                 parameters_q = {
                     k: v

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/src/iqm/pulse/gates/__init__.py

@@ -184,13 +184,13 @@ def _validate_operation(
     """
     if not overwrite and gate_name in operations:
         old = operations.get(gate_name)
-        same = _compare_operations(new_op, old)
+        same = _compare_operations(new_op, old)  # type: ignore[arg-type]
         if not same:
             raise ValueError(f"{gate_name} already registered with different parameters")
 
     if gate_name in _quantum_ops_library:
         default = _quantum_ops_library.get(gate_name)
-        same = _compare_operations(new_op, default)
+        same = _compare_operations(new_op, default)  # type: ignore[arg-type]
         if not same:
             raise ValueError(f"{gate_name} conflicts with a canonical operation in iqm-pulse")
 
@@ -198,7 +198,7 @@ def _validate_operation(
         if gate_name in operations and operations[gate_name].unitary is not None:
             unitary = operations[gate_name].unitary
         elif gate_name in _quantum_ops_library and _quantum_ops_library[gate_name].unitary is not None:
-            unitary = default.unitary
+            unitary = default.unitary  # type: ignore[union-attr]
         else:
             unitary = None
         new_op = replace(new_op, unitary=unitary)
@@ -239,9 +239,9 @@ def _register_gate(
         }
         if quantum_op_specs:
             new_kwargs |= quantum_op_specs
-            new_kwargs["params"] = tuple(new_kwargs.get("params", ()))
+            new_kwargs["params"] = tuple(new_kwargs.get("params", ()))  # type: ignore[arg-type]
 
-        new_op = QuantumOp(**new_kwargs)
+        new_op = QuantumOp(**new_kwargs)  # type: ignore[arg-type]  # type: ignore[arg-type]  # type: ignore[arg-type]  # type: ignore[arg-type]  # type: ignore[arg-type]  # type: ignore[arg-type]  # type: ignore[arg-type]
 
     return new_op
 
@@ -275,7 +275,7 @@ def _add_implementation(
     if not get_implementation_class(impl_class.__name__):
         expose_implementation(impl_class, overwrite)
 
-    return new_op
+    return new_op  # type: ignore[return-value]
 
 
 def _validate_implementation(

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/src/iqm/pulse/gates/conditional.py

@@ -70,10 +70,10 @@ class CCPRX_Composite(CompositeGate):
         qubit = self.locus[0]
         awg_name = self.builder.get_drive_channel(qubit)
 
-        prx_gate: PRX_SinglePulse_GateImplementation = self.build("prx", self.locus)
+        prx_gate: PRX_SinglePulse_GateImplementation = self.build("prx", self.locus)  # type: ignore[assignment]
         # FIXME assumes PRX gates only use this one implementation as the default,
         # with just a drive channel and a single IQPulse.
-        pulse = prx_gate(angle, phase).atom[prx_gate.channel][0]
+        pulse = prx_gate(angle, phase).atom[prx_gate.channel][0]  # type: ignore[union-attr,index]
         wait = Wait(pulse.duration)  # idling, can be replaced with a DD sequence later on
 
         # TODO: use the actual inputted label when the HW supports many labels per drive channel

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/src/iqm/pulse/gates/cz.py

@@ -81,7 +81,7 @@ class FluxPulseGate(GateImplementation):
     """Flux pulse Waveform to be played in the qubit flux AWG."""
     root_parameters: dict[str, Parameter | Setting] = {
         "duration": Parameter("", "Gate duration", "s"),
-        "rz": {
+        "rz": {  # type: ignore[dict-item]
             "*": Parameter("", "Z rotation angle", "rad"),  # wildcard parameter
         },
     }
@@ -108,7 +108,7 @@ class FluxPulseGate(GateImplementation):
             flux_channel = builder.get_flux_channel(component_name)
             params = self.convert_calibration_data(
                 calibration_data[cal_node_name],
-                self.parameters[cal_node_name],
+                self.parameters[cal_node_name],  # type: ignore[arg-type]
                 builder.channels[flux_channel],
                 duration,
             )
@@ -266,7 +266,7 @@ class CouplerFluxPulseQubitACStarkPulseGate(GateImplementation):
 
     root_parameters: dict[str, Parameter | Setting] = {
         "duration": Parameter("", "Gate duration", "s"),
-        "rz": {
+        "rz": {  # type: ignore[dict-item]
             "*": Parameter("", "Z rotation angle", "rad"),
         },
     }
@@ -294,7 +294,7 @@ class CouplerFluxPulseQubitACStarkPulseGate(GateImplementation):
             flux_channel = builder.get_flux_channel(component_name)
             params = self.convert_calibration_data(
                 calibration_data[cal_node_name],
-                self.parameters[cal_node_name],
+                self.parameters[cal_node_name],  # type: ignore[arg-type]
                 builder.channels[flux_channel],
                 duration,
             )
@@ -312,7 +312,7 @@ class CouplerFluxPulseQubitACStarkPulseGate(GateImplementation):
             drive_channel = builder.get_drive_channel(component_name)
             params = self.convert_calibration_data(
                 calibration_data[cal_node_name],
-                self.parameters[cal_node_name],
+                self.parameters[cal_node_name],  # type: ignore[arg-type]
                 builder.channels[drive_channel],
                 duration,
             )

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/src/iqm/pulse/gates/default_gates.py

@@ -115,95 +115,98 @@ _default_operations: QuantumOpTable = {
         QuantumOp(
             "barrier",
             0,
-            implementations=_default_implementations["barrier"],
+            implementations=_default_implementations["barrier"],  # type: ignore[arg-type]
             symmetric=True,
         ),
         QuantumOp(
             "delay",
             0,
             ("duration",),
-            implementations=_default_implementations["delay"],
+            implementations=_default_implementations["delay"],  # type: ignore[arg-type]
             symmetric=True,
         ),
         QuantumOp(
             "measure",
             0,
             ("key",),
-            implementations=_default_implementations["measure"],
+            implementations=_default_implementations["measure"],  # type: ignore[arg-type]
             factorizable=True,
         ),
         QuantumOp(
             "prx",
             1,
             ("angle", "phase"),
-            implementations=_default_implementations["prx"],
+            implementations=_default_implementations["prx"],  # type: ignore[arg-type]
             unitary=get_unitary_prx,
         ),
         QuantumOp(
             "prx_12",
             1,
             ("angle", "phase"),
-            implementations=_default_implementations["prx_12"],
+            implementations=_default_implementations["prx_12"],  # type: ignore[arg-type]
         ),
         QuantumOp(
             "u",
             1,
             ("theta", "phi", "lam"),
-            implementations=_default_implementations["u"],
+            implementations=_default_implementations["u"],  # type: ignore[arg-type]
             unitary=get_unitary_u,
         ),
         QuantumOp(
-            "sx", 1, implementations=_default_implementations["sx"], unitary=lambda: get_unitary_prx(np.pi / 2, 0)
+            "sx",
+            1,
+            implementations=_default_implementations["sx"],  # type: ignore[arg-type]
+            unitary=lambda: get_unitary_prx(np.pi / 2, 0),
         ),
         QuantumOp(
             "rz",
             1,
             ("angle",),
-            implementations=_default_implementations["rz"],
+            implementations=_default_implementations["rz"],  # type: ignore[arg-type]
             unitary=get_unitary_rz,
         ),
         QuantumOp(
             "rz_physical",
             1,
-            implementations=_default_implementations["rz_physical"],
+            implementations=_default_implementations["rz_physical"],  # type: ignore[arg-type]
         ),
         QuantumOp(
             "cz",
             2,
             (),
-            implementations=_default_implementations["cz"],
+            implementations=_default_implementations["cz"],  # type: ignore[arg-type]
             symmetric=True,
             unitary=lambda: np.diag([1.0, 1.0, 1.0, -1.0]),
         ),
         QuantumOp(
             "move",
             2,
-            implementations=_default_implementations["move"],
+            implementations=_default_implementations["move"],  # type: ignore[arg-type]
         ),
         QuantumOp(
             "cc_prx",
             1,
             ("angle", "phase", "feedback_qubit", "feedback_key"),
-            implementations=_default_implementations["cc_prx"],
+            implementations=_default_implementations["cc_prx"],  # type: ignore[arg-type]
         ),
         QuantumOp(
             "reset",
             0,
-            implementations=_default_implementations["reset"],
+            implementations=_default_implementations["reset"],  # type: ignore[arg-type]
             symmetric=True,
             factorizable=True,
         ),
         QuantumOp(
             "reset_wait",
             0,
-            implementations=_default_implementations["reset_wait"],
+            implementations=_default_implementations["reset_wait"],  # type: ignore[arg-type]
             symmetric=True,
             factorizable=True,
         ),
         QuantumOp(
             "flux_multiplexer",
             0,
-            implementations=_default_implementations["flux_multiplexer"],
+            implementations=_default_implementations["flux_multiplexer"],  # type: ignore[arg-type]
         ),
     ]
 }
@@ -215,95 +218,98 @@ _quantum_ops_library = {
         QuantumOp(
             "barrier",
             0,
-            implementations=_default_implementations["barrier"],
+            implementations=_default_implementations["barrier"],  # type: ignore[arg-type]
             symmetric=True,
         ),
         QuantumOp(
             "delay",
             0,
             ("duration",),
-            implementations=_default_implementations["delay"],
+            implementations=_default_implementations["delay"],  # type: ignore[arg-type]
             symmetric=True,
         ),
         QuantumOp(
             "measure",
             0,
             ("key",),
-            implementations=_default_implementations["measure"],
+            implementations=_default_implementations["measure"],  # type: ignore[arg-type]
             factorizable=True,
         ),
         QuantumOp(
             "prx",
             1,
             ("angle", "phase"),
-            implementations=_default_implementations["prx"],
+            implementations=_default_implementations["prx"],  # type: ignore[arg-type]
             unitary=get_unitary_prx,
         ),
         QuantumOp(
             "prx_12",
             1,
             ("angle", "phase"),
-            implementations=_default_implementations["prx_12"],
+            implementations=_default_implementations["prx_12"],  # type: ignore[arg-type]
         ),
         QuantumOp(
             "u",
             1,
             ("theta", "phi", "lam"),
-            implementations=_default_implementations["u"],
+            implementations=_default_implementations["u"],  # type: ignore[arg-type]
             unitary=get_unitary_u,
         ),
         QuantumOp(
-            "sx", 1, implementations=_default_implementations["sx"], unitary=lambda: get_unitary_prx(np.pi / 2, 0)
+            "sx",
+            1,
+            implementations=_default_implementations["sx"],  # type: ignore[arg-type]
+            unitary=lambda: get_unitary_prx(np.pi / 2, 0),
         ),
         QuantumOp(
             "rz",
             1,
             ("angle",),
-            implementations=_default_implementations["rz"],
+            implementations=_default_implementations["rz"],  # type: ignore[arg-type]
             unitary=get_unitary_rz,
         ),
         QuantumOp(
             "rz_physical",
             1,
-            implementations=_default_implementations["rz_physical"],
+            implementations=_default_implementations["rz_physical"],  # type: ignore[arg-type]
         ),
         QuantumOp(
             "cz",
             2,
             (),
-            implementations=_default_implementations["cz"],
+            implementations=_default_implementations["cz"],  # type: ignore[arg-type]
             symmetric=True,
             unitary=lambda: np.diag([1.0, 1.0, 1.0, -1.0]),
         ),
         QuantumOp(
             "move",
             2,
-            implementations=_default_implementations["move"],
+            implementations=_default_implementations["move"],  # type: ignore[arg-type]
         ),
         QuantumOp(
             "cc_prx",
             1,
             ("angle", "phase", "feedback_qubit", "feedback_key"),
-            implementations=_default_implementations["cc_prx"],
+            implementations=_default_implementations["cc_prx"],  # type: ignore[arg-type]
         ),
         QuantumOp(
             "reset",
             0,
-            implementations=_default_implementations["reset"],
+            implementations=_default_implementations["reset"],  # type: ignore[arg-type]
             symmetric=True,
             factorizable=True,
         ),
         QuantumOp(
             "reset_wait",
             0,
-            implementations=_default_implementations["reset_wait"],
+            implementations=_default_implementations["reset_wait"],  # type: ignore[arg-type]
             symmetric=True,
             factorizable=True,
         ),
         QuantumOp(
             "flux_multiplexer",
             0,
-            implementations=_default_implementations["flux_multiplexer"],
+            implementations=_default_implementations["flux_multiplexer"],  # type: ignore[arg-type]
         ),
     ]
 }

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/src/iqm/pulse/gates/flux_multiplexer.py

@@ -122,7 +122,7 @@ class FluxMultiplexer_SampleLinear(GateImplementation):
                         multiple = int(inst.duration / real_pulse_dict["duration"])
                         chopped_seg.extend(multiple * [Wait(real_pulse_dict["duration"])])
                     else:
-                        chopped_seg.append(inst)
+                        chopped_seg.append(inst)  # type: ignore[arg-type]
                 scheduled_fluxes[channel] = Segment(chopped_seg)
         max_flux_depth = next(len(seg) for ch, seg in scheduled_fluxes.items() if ch in crosstalk_channels)
         # pad the relevant flux channels not present in the schedule at all with correct number of wait tiles
@@ -157,7 +157,7 @@ class FluxMultiplexer_SampleLinear(GateImplementation):
                 if max(abs(multiplexed_samples)) < TOLERANCE:
                     multiplexed_pulse = Wait(real_pulse_dict["duration"])
                 else:
-                    multiplexed_pulse = RealPulse(
+                    multiplexed_pulse = RealPulse(  # type: ignore[assignment]
                         scale=1.0, duration=real_pulse_dict["duration"], wave=Samples(multiplexed_samples)
                     )
                 corrected_fluxes[flux_channel]._instructions[layer] = multiplexed_pulse

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/src/iqm/pulse/gates/measure.py

@@ -106,7 +106,7 @@ class Measure_CustomWaveforms(CustomIQWaveforms):
         """Cache for :meth:`time_trace`."""
 
         if len(locus) == 1:  # factorizable gates only need calibration on 1-loci
-            self._probe_line: ProbeChannelProperties = self.builder.channels[
+            self._probe_line: ProbeChannelProperties = self.builder.channels[  # type: ignore[assignment]
                 self.builder.get_probe_channel(self.locus[0])
             ]
             c_freq = self._probe_line.center_frequency
@@ -194,7 +194,7 @@ class Measure_CustomWaveforms(CustomIQWaveforms):
                 modulation_frequency=if_freq,  # TODO: should be fixed to -if_freq in Programmable RO Phase2?
             )
 
-        acquisition_type = root_params.get("acquisition_type", self.root_parameters["acquisition_type"].value)
+        acquisition_type = root_params.get("acquisition_type", self.root_parameters["acquisition_type"].value)  # type: ignore[union-attr]
         acquisition_delay = round(self._probe_line.duration_to_samples(root_params["acquisition_delay"]))
         acquisition_label = "TO_BE_REPLACED"
         if acquisition_type == "complex":
@@ -245,7 +245,7 @@ class Measure_CustomWaveforms(CustomIQWaveforms):
                     # TODO: use the actual ``feedback_key`` when AWGs support multiple feedback labels
                     feedback_bit = f"{self.locus[0]}__{FEEDBACK_KEY}"
                     replacements["feedback_signal_label"] = feedback_bit
-                acquisitions = (replace(self._acquisition_method, **replacements),) if do_acquisition else ()
+                acquisitions = (replace(self._acquisition_method, **replacements),) if do_acquisition else ()  # type: ignore[arg-type]
                 multiplexed_iq = MultiplexedIQPulse(
                     duration=self._probe_instruction.duration + self._probe_line.integration_start_dead_time,
                     entries=((self._probe_instruction, self._probe_line.integration_start_dead_time),),
@@ -289,7 +289,7 @@ class Measure_CustomWaveforms(CustomIQWaveforms):
                 # _skip_override used for child classes build on `Measure_CustomWaveforms` to not call `.probe_timebox`
                 # from the parent class, but from this class instead.
                 probe_timeboxes = [
-                    self.builder.get_implementation(
+                    self.builder.get_implementation(  # type: ignore[attr-defined]
                         self.parent.name, (c,), impl_name=self.name, priority_calibration=self._prio_calibration
                     ).probe_timebox(key, feedback_key, do_acquisition, _skip_override=True)
                     for c in self.locus
@@ -380,7 +380,7 @@ class Measure_CustomWaveforms(CustomIQWaveforms):
         # additional caching for time traces since the acquisitions differ from the ones in _call
         if args not in self._time_traces:
             probe_timebox = deepcopy(self.probe_timebox(key=key, feedback_key=feedback_key, _skip_override=True))
-            for probe_channel, segment in probe_timebox.atom.items():
+            for probe_channel, segment in probe_timebox.atom.items():  # type: ignore[union-attr]
                 readout_trigger = segment[0]
                 # TODO instead of editing the probe_timebox output contents, we should make the function itself do this
                 # so we would not need to blindly search through the channels
@@ -398,7 +398,8 @@ class Measure_CustomWaveforms(CustomIQWaveforms):
                     duration_samples = probe_line.duration_to_int_samples(acquisition_duration)
                 else:
                     duration_samples = max(
-                        acq.weights.duration + acq.delay_samples - delay_samples for acq in readout_trigger.acquisitions
+                        acq.weights.duration + acq.delay_samples - delay_samples  # type: ignore[attr-defined]
+                        for acq in readout_trigger.acquisitions
                     )
                 label_key = key or DEFAULT_TIME_TRACE_KEY
                 time_trace = TimeTrace(
@@ -417,7 +418,7 @@ class Measure_CustomWaveforms(CustomIQWaveforms):
     def duration_in_seconds(self) -> float:
         probe_timebox = self.probe_timebox()
         readout_schedule = probe_timebox.atom
-        return readout_schedule.duration_in_seconds(self.builder.channels)
+        return readout_schedule.duration_in_seconds(self.builder.channels)  # type: ignore[union-attr]
 
     @classmethod
     def get_locus_mapping_name(cls, operation_name: str, implementation_name: str) -> str:
@@ -464,7 +465,7 @@ class ProbePulse_CustomWaveforms(CustomIQWaveforms):
         self, parent: QuantumOp, name: str, locus: Locus, calibration_data: OILCalibrationData, builder: ScheduleBuilder
     ):
         super().__init__(parent, name, locus, calibration_data, builder)
-        self._probe_line: ProbeChannelProperties = self.builder.channels[
+        self._probe_line: ProbeChannelProperties = self.builder.channels[  # type: ignore[assignment]
             self.builder.component_channels[self.locus[0]]["readout"]
         ]
         self._duration = (
@@ -568,9 +569,9 @@ class ProbePulse_CustomWaveforms(CustomIQWaveforms):
         return final_box
 
     def duration_in_seconds(self) -> float:
-        probe_timebox = self().children[0]
+        probe_timebox = self().children[0]  # type: ignore[union-attr]
         probe_channel = self.builder.component_channels[self.locus[0]]["readout"]
-        return self.builder.channels[probe_channel].duration_to_seconds(probe_timebox.atom.duration)
+        return self.builder.channels[probe_channel].duration_to_seconds(probe_timebox.atom.duration)  # type: ignore[union-attr]
 
     @classmethod
     def get_locus_mapping_name(cls, operation_name: str, implementation_name: str) -> str:
@@ -601,7 +602,7 @@ class ProbePulse_CustomWaveforms_noIntegration(CustomIQWaveforms):
         """Cache for :meth:`probe_timebox`."""
 
         if len(locus) == 1:  # factorizable gates only need calibration on 1-loci
-            self._probe_line: ProbeChannelProperties = self.builder.channels[
+            self._probe_line: ProbeChannelProperties = self.builder.channels[  # type: ignore[assignment]
                 self.builder.get_probe_channel(self.locus[0])
             ]
             c_freq = self._probe_line.center_frequency
@@ -723,7 +724,7 @@ class ProbePulse_CustomWaveforms_noIntegration(CustomIQWaveforms):
                     )
             else:
                 probe_timeboxes = [
-                    self.builder.get_implementation(
+                    self.builder.get_implementation(  # type: ignore[attr-defined]
                         self.parent.name, (c,), impl_name=self.name, priority_calibration=self._prio_calibration
                     ).probe_timebox(key, feedback_key, do_acquisition)
                     for c in self.locus
@@ -775,7 +776,7 @@ class ProbePulse_CustomWaveforms_noIntegration(CustomIQWaveforms):
     def duration_in_seconds(self) -> float:
         probe_timebox = self.probe_timebox()
         readout_schedule = probe_timebox.atom
-        return readout_schedule.duration_in_seconds(self.builder.channels)
+        return readout_schedule.duration_in_seconds(self.builder.channels)  # type: ignore[union-attr]
 
     @classmethod
     def get_locus_mapping_name(cls, operation_name: str, implementation_name: str) -> str:
@@ -832,12 +833,13 @@ class Shelved_Measure_CustomWaveforms(Measure_CustomWaveforms):
         multiplexed_timeboxes = super().probe_timebox(key, feedback_key)
         prx_12_impl = [self.builder.get_implementation("prx_12", [q])(np.pi) for q in self.locus]
 
-        boxes = prx_12_impl + multiplexed_timeboxes + prx_12_impl
+        boxes = prx_12_impl + multiplexed_timeboxes + prx_12_impl  # type: ignore[operator]
         return boxes
 
     def _call(self, key: str = "", feedback_key: str = "") -> TimeBox:  # type: ignore[override]
         shelved_measure_box = TimeBox.composite(
-            self.probe_timebox(key=key, feedback_key=feedback_key), label=f"Readout on {self.locus}"
+            self.probe_timebox(key=key, feedback_key=feedback_key),  # type: ignore[arg-type]
+            label=f"Readout on {self.locus}",
         )
         shelved_measure_box.neighborhood_components[0] = shelved_measure_box.children[
             len(self.locus)

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/src/iqm/pulse/gates/move.py

@@ -104,7 +104,7 @@ class MOVE_CustomWaveforms(FluxPulseGate):
 
     root_parameters: dict[str, Parameter | Setting] = {
         "duration": Parameter("", "Gate duration", "s"),
-        "rz": {
+        "rz": {  # type: ignore[dict-item]
             "*": Parameter("", "Z rotation angle", "rad"),  # wildcard parameter
         },
         "detuning": Parameter("", "Qubit - resonator detuning", "Hz"),

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/src/iqm/pulse/gates/prx.py

@@ -120,8 +120,8 @@ class PrxGateImplementation(GateImplementation):
 
         """
         box = self(angle=angle, phase=0)
-        box.label = f"Rx on {self.locus[0]}"
-        return box
+        box.label = f"Rx on {self.locus[0]}"  # type: ignore[union-attr]
+        return box  # type: ignore[return-value]
 
     def ry(self, angle: float) -> TimeBox:
         """Y rotation gate.
@@ -134,8 +134,8 @@ class PrxGateImplementation(GateImplementation):
 
         """
         box = self(angle=angle, phase=np.pi / 2)
-        box.label = f"Ry on {self.locus[0]}"
-        return box
+        box.label = f"Ry on {self.locus[0]}"  # type: ignore[union-attr]
+        return box  # type: ignore[return-value]
 
     def clifford(self, xy_gate: XYGate) -> TimeBox:
         """One-qubit XY Clifford gates.
@@ -197,12 +197,12 @@ class PRX_SinglePulse_GateImplementation(SinglePulseGate, PrxGateImplementation)
     duration being zero, the gate implementation will apply a ``Block(0)`` instruction to the qubit's drive channel.
     """
 
-    def _call(self, angle: float, phase: float = 0.0) -> TimeBox:
+    def _call(self, angle: float, phase: float = 0.0) -> TimeBox:  # type: ignore[override]
         scale, new_phase = _normalize_params(angle, phase)
         pulse = self.pulse.copy(
-            scale_i=scale * self.pulse.scale_i,
-            scale_q=scale * self.pulse.scale_q,
-            phase=self.pulse.phase + new_phase,
+            scale_i=scale * self.pulse.scale_i,  # type: ignore[attr-defined]
+            scale_q=scale * self.pulse.scale_q,  # type: ignore[attr-defined]
+            phase=self.pulse.phase + new_phase,  # type: ignore[attr-defined]
         )
         if self.pulse.duration > TOLERANCE:
             timebox = self.to_timebox(Schedule({self.channel: [pulse]}))
@@ -214,7 +214,7 @@ class PRX_SinglePulse_GateImplementation(SinglePulseGate, PrxGateImplementation)
     @property
     def iq_pulse(self) -> IQPulse:
         """Alias for ``self.pulse`` for backward compatibility"""
-        return self.pulse
+        return self.pulse  # type: ignore[return-value]
 
 
 class PRX_CustomWaveforms(PRX_SinglePulse_GateImplementation, CustomIQWaveforms):
@@ -317,12 +317,12 @@ class PRX_CustomWaveformsSX(PRX_SinglePulse_GateImplementation, CustomIQWaveform
             timebox = self.to_timebox(Schedule({self.channel: [pulse]}))
         elif np.isclose(new_angle, np.pi / 2, atol=1e-8):
             pulse = self.pulse.copy(
-                phase=self.pulse.phase + new_phase,
+                phase=self.pulse.phase + new_phase,  # type: ignore[attr-defined]
             )
             timebox = self.to_timebox(Schedule({self.channel: [pulse]}))
         elif np.isclose(new_angle, np.pi, atol=1e-8):
             pulse = self.pulse.copy(
-                phase=self.pulse.phase + new_phase,
+                phase=self.pulse.phase + new_phase,  # type: ignore[attr-defined]
             )
             timebox = self.to_timebox(Schedule({self.channel: [pulse, pulse]}))
         else:
@@ -332,12 +332,12 @@ class PRX_CustomWaveformsSX(PRX_SinglePulse_GateImplementation, CustomIQWaveform
             phase_1, phase_increment_1 = phase_transformation(rz_a, 0)
             phase_2, phase_increment_2 = phase_transformation(rz_b, rz_c)
             pulse_1 = self.pulse.copy(
-                phase=normalize_angle(self.pulse.phase + phase_1),
-                phase_increment=normalize_angle(self.pulse.phase_increment + phase_increment_1),
+                phase=normalize_angle(self.pulse.phase + phase_1),  # type: ignore[attr-defined]
+                phase_increment=normalize_angle(self.pulse.phase_increment + phase_increment_1),  # type: ignore[attr-defined]
             )
             pulse_2 = self.pulse.copy(
-                phase=normalize_angle(self.pulse.phase + phase_2),
-                phase_increment=normalize_angle(self.pulse.phase_increment + phase_increment_2),
+                phase=normalize_angle(self.pulse.phase + phase_2),  # type: ignore[attr-defined]
+                phase_increment=normalize_angle(self.pulse.phase_increment + phase_increment_2),  # type: ignore[attr-defined]
             )
             timebox = self.to_timebox(Schedule({self.channel: [pulse_1, pulse_2]}))
 
@@ -482,14 +482,14 @@ class ABC_Constant_smooth(PrxGateImplementation):
         params_for_stark["n_samples"] = max(
             int(round(params_for_stark["n_samples"] * (1 - 2 * params_for_stark["rise_time"]), 0)), 0
         )
-        self.main_waveform = self._main_pulse(**params_for_stark)
+        self.main_waveform = self._main_pulse(**params_for_stark)  # type: ignore[assignment]
 
         params_for_risefall["n_samples"] = int(
             round(params_for_risefall["n_samples"] * params_for_risefall["rise_time"], 0)
         )
 
-        self.fall_waveform = self._fall_pulse(**params_for_risefall)
-        self.rise_waveform = self._rise_pulse(**params_for_risefall)
+        self.fall_waveform = self._fall_pulse(**params_for_risefall)  # type: ignore[assignment]
+        self.rise_waveform = self._rise_pulse(**params_for_risefall)  # type: ignore[assignment]
 
         self.channel = drive_channel
         self.special_implementation = True  # DEBUG
@@ -513,7 +513,7 @@ class ABC_Constant_smooth(PrxGateImplementation):
 
         parameters["rise_time"] = Parameter("", "gate rise time", "s")
 
-        cls.parameters = {
+        cls.parameters = {  # type: ignore[assignment]
             "duration": Parameter("", "Gate duration", "s"),
         } | parameters
 

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/src/iqm/pulse/gates/reset.py

@@ -64,11 +64,11 @@ class Reset_Conditional(CompositeGate):
         )
         # try to get a qnd measurement, otherwise use default one
         try:  # TODO: should the QND measurement be its own QuantumOp instead?
-            probe_timebox = self.build("measure", resettable, impl_name="constant_qnd").probe_timebox(
+            probe_timebox = self.build("measure", resettable, impl_name="constant_qnd").probe_timebox(  # type: ignore[attr-defined]
                 RESET_MEASUREMENT_KEY, feedback_key=RESET_FEEDBACK_KEY
             )
         except (ValueError, KeyError):
-            probe_timebox = self.build("measure", resettable).probe_timebox(
+            probe_timebox = self.build("measure", resettable).probe_timebox(  # type: ignore[attr-defined]
                 RESET_MEASUREMENT_KEY, feedback_key=RESET_FEEDBACK_KEY
             )
         virtual_channels = set()
@@ -89,7 +89,7 @@ class Reset_Conditional(CompositeGate):
         probe_timebox.neighborhood_components[0] = blocks
         measure = TimeBox.composite([probe_timebox])
         measure.neighborhood_components[0] = blocks
-        return TimeBox.composite([measure, resets])
+        return TimeBox.composite([measure, resets])  # type: ignore[list-item]
 
     @classmethod
     def get_locus_mapping_name(cls, operation_name: str, implementation_name: str) -> str:
@@ -123,7 +123,7 @@ class Reset_Wait(GateImplementation):
             prio_calibration = self.calibration_data if self.calibration_data else None
             waits_box = TimeBox.composite(
                 [
-                    self.builder.get_implementation(
+                    self.builder.get_implementation(  # type: ignore[attr-defined]
                         self.parent.name, (q,), impl_name=self.name, priority_calibration=prio_calibration
                     ).wait_box()
                     for q in self.locus

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/src/iqm/pulse/gates/rz.py

@@ -144,7 +144,7 @@ class RZ_Virtual(GateImplementation):
         timebox.neighborhood_components[0] = set(self.locus)
         return timebox
 
-    parameters: dict[str, Parameter] = {}
+    parameters: dict[str, Parameter] = {}  # type: ignore[assignment]
 
     def duration_in_seconds(self) -> float:
         return self.duration
@@ -191,7 +191,7 @@ class RZ_ACStarkShift(GateImplementation):
         parameters["amplitude"] = Parameter("", "amplitude", "")
         parameters["phase_increment"] = Parameter("", "phase increment", "rad")
 
-        cls.parameters = {
+        cls.parameters = {  # type: ignore[assignment]
             "duration": Parameter("", "Gate duration", "s"),
         } | parameters
 
@@ -234,7 +234,7 @@ class RZ_ACStarkShift_CosineRiseFall(RZ_ACStarkShift, ac_stark_waveform=Modulate
     """AC stark pulse implemented as a modulated cosine rise fall pulse."""
 
 
-class RZ_ACStarkShift_smoothConstant(
+class RZ_ACStarkShift_smoothConstant(  # type: ignore[call-arg]  # type: ignore[call-arg]  # type: ignore[call-arg]
     Constant_PRX_with_smooth_rise_fall,
     rise_waveform=CosineRise,  # type:ignore[call-arg]
     main_waveform=Constant,  # type:ignore[call-arg]
@@ -260,8 +260,8 @@ class RZ_PRX_Composite(CompositeGate):
         prx = self.build("prx", self.locus)
         return TimeBox.composite(
             [
-                prx.ry(np.pi / 2),
-                prx.rx(angle),
-                prx.ry(-np.pi / 2),
+                prx.ry(np.pi / 2),  # type: ignore[attr-defined]
+                prx.rx(angle),  # type: ignore[attr-defined]
+                prx.ry(-np.pi / 2),  # type: ignore[attr-defined]
             ]
         )

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/src/iqm/pulse/gates/sx.py

@@ -42,6 +42,6 @@ class SXGate(CompositeGate):
 
     def _call(self) -> TimeBox:  # type: ignore[override]
         """Call PRX gate with angle equals to pi / 2."""
-        prx_gate: PRX_SinglePulse_GateImplementation | PRX_CustomWaveformsSX = self.build("prx", self.locus)
-        pulse = prx_gate(np.pi / 2, 0.0).atom[prx_gate.channel][0]
+        prx_gate: PRX_SinglePulse_GateImplementation | PRX_CustomWaveformsSX = self.build("prx", self.locus)  # type: ignore[assignment]
+        pulse = prx_gate(np.pi / 2, 0.0).atom[prx_gate.channel][0]  # type: ignore[union-attr,index]
         return self.to_timebox(Schedule({prx_gate.channel: [pulse]}))

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/src/iqm/pulse/gates/u.py

@@ -103,7 +103,9 @@ class UGate(CompositeGate):
         # the PRX implementation. This isn't safe in general, can we find a better solution?
 
         prx_gate = self.build("prx", self.locus)
-        pulse_train = prx_gate(theta, np.pi / 2).atom[prx_gate.channel]  # RY pulse
+        pulse_train = prx_gate(theta, np.pi / 2).atom[  # type: ignore[union-attr]
+            prx_gate.channel  # type: ignore[index,attr-defined]
+        ]  # RY pulse
 
         # Check if the pulse train have one or several pulses.
         if len(pulse_train) == 1:
@@ -116,7 +118,7 @@ class UGate(CompositeGate):
                 phase=normalize_angle(pulse.phase + new_phase),
                 phase_increment=normalize_angle(pulse.phase_increment + new_phase_increment),
             )
-            timebox = self.to_timebox(Schedule({prx_gate.channel: [new_pulse]}))
+            timebox = self.to_timebox(Schedule({prx_gate.channel: [new_pulse]}))  # type: ignore[attr-defined]
 
         else:
             # Assumes the PRX pulse train begins and ends with IQPulses.
@@ -139,7 +141,7 @@ class UGate(CompositeGate):
             )
             other_pulses = [pulse.copy() for pulse in pulse_train[1:-1]]
             new_pulses = [new_pulse_a] + other_pulses + [new_pulse_b]
-            timebox = self.to_timebox(Schedule({prx_gate.channel: new_pulses}))
+            timebox = self.to_timebox(Schedule({prx_gate.channel: new_pulses}))  # type: ignore[attr-defined]
 
         timebox.neighborhood_components[0] = set(self.locus)
         return timebox

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/src/iqm/pulse/quantum_ops.py

@@ -229,7 +229,7 @@ def validate_op_calibration(calibration: OpCalibrationDataTree, ops: QuantumOpTa
 
             default_cal_data = loci.get((), {})
             for locus, cal_data in loci.items():
-                validate_locus_calibration(merge_dicts(default_cal_data, cal_data), impl, op, impl_name, locus)
+                validate_locus_calibration(merge_dicts(default_cal_data, cal_data), impl, op, impl_name, locus)  # type: ignore[arg-type]
 
 
 def validate_locus_calibration(

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/src/iqm/pulse/scheduler.py

@@ -266,8 +266,8 @@ def extend_schedule_new(  # noqa: PLR0915
             _ = channels[ch].duration_to_int_samples(duration, message=f"{ch}: {iA} + {iB}: overlap duration")
 
         if isinstance(iA, Nothing) and isinstance(iB, Nothing):
-            return Nothing(duration=duration)
-        return Block(duration=duration)
+            return Nothing(duration=duration)  # type: ignore[arg-type]
+        return Block(duration=duration)  # type: ignore[arg-type]
 
     def find_start_time() -> float:
         """Find the earliest possible start time for schedule B."""
@@ -297,7 +297,7 @@ def extend_schedule_new(  # noqa: PLR0915
         if distance >= -TOL:
             if distance > TOL:
                 # gap, add a Nothing between A and B
-                segment_A._instructions.append(Nothing(duration=distance))
+                segment_A._instructions.append(Nothing(duration=distance))  # type: ignore[arg-type]
             # add B
             segment_A._instructions.extend(pointer_B.tail())
         else:

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/src/iqm/pulse/timebox.py

@@ -378,7 +378,7 @@ class MultiplexedProbeTimeBox(TimeBox):
         """
         schedule = {probe_channel: [readout_trigger]}
         for channel in block_channels:
-            schedule[channel] = [Block(block_duration)]
+            schedule[channel] = [Block(block_duration)]  # type: ignore[list-item]
         box = MultiplexedProbeTimeBox(
             label=label,
             locus_components=set(locus_components),

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0/src/iqm_pulse.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: iqm-pulse
-Version: 9.19.0
+Version: 9.21.0
 Summary: A Python-based project for providing interface and implementations for control pulses.
 Author-email: IQM Finland Oy <info@meetiqm.com>
 License:                                  Apache License
@@ -219,6 +219,7 @@ Requires-Dist: python-rapidjson==1.20
 Requires-Dist: jinja2==3.0.3
 Requires-Dist: numpy<3.0,>=1.26.4
 Requires-Dist: scipy<1.16,>=1.11.4
+Requires-Dist: scipy-stubs
 
 IQM Pulse library
 =================

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/src/iqm_pulse.egg-info/SOURCES.txt

@@ -34,6 +34,7 @@ src/iqm/pulse/base_utils.py
 src/iqm/pulse/builder.py
 src/iqm/pulse/circuit_operations.py
 src/iqm/pulse/gate_implementation.py
+src/iqm/pulse/py.typed
 src/iqm/pulse/quantum_ops.py
 src/iqm/pulse/scheduler.py
 src/iqm/pulse/timebox.py

{iqm_pulse-9.19.0 → iqm_pulse-9.21.0}/src/iqm_pulse.egg-info/requires.txt

@@ -4,3 +4,4 @@ python-rapidjson==1.20
 jinja2==3.0.3
 numpy<3.0,>=1.26.4
 scipy<1.16,>=1.11.4
+scipy-stubs

iqm_pulse-9.21.0/version.txt

@@ -0,0 +1 @@
+9.21.0

iqm_pulse-9.19.0/version.txt

@@ -1 +0,0 @@
-9.19.0