iqm-pulla 8.1.0__tar.gz → 8.3.0__tar.gz

{iqm_pulla-8.1.0 → iqm_pulla-8.3.0}/CHANGELOG.rst

@@ -2,6 +2,22 @@
 Changelog
 =========
 
+Version 8.3.0 (2025-07-09)
+==========================
+
+Features
+--------
+
+- Enable mypy type checking in CI and add temporary type ignores to the source code. :issue:`SW-1615`
+
+Version 8.2.0 (2025-07-02)
+==========================
+
+Bug fixes
+---------
+
+- Fix type errors raised by mypy.
+
 Version 8.1.0 (2025-06-18)
 ==========================
 

{iqm_pulla-8.1.0 → iqm_pulla-8.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: iqm-pulla
-Version: 8.1.0
+Version: 8.3.0
 Summary: Client library for pulse-level access to an IQM quantum computer
 Author-email: IQM Finland Oy <developers@meetiqm.com>
 License:                                  Apache License

{iqm_pulla-8.1.0 → iqm_pulla-8.3.0}/pyproject.toml

@@ -21,6 +21,14 @@ file = "LICENSE.txt"
 Documentation = "https://iqm-finland.github.io/docs/iqm-pulla/"
 Homepage = "https://pypi.org/project/iqm-pulla/"
 
+[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_pulla-8.1.0 → iqm_pulla-8.3.0}/src/iqm/cpc/compiler/dd.py

@@ -515,7 +515,7 @@ def insert_dd_sequences(
         wait_duration = wait.duration
 
         # NOTE: Duration is always calculated using "X"
-        prx_duration = _get_channel_segment(builder, prx.rx(math.pi), channel_name).duration
+        prx_duration = _get_channel_segment(builder, prx.rx(math.pi), channel_name).duration  # type: ignore[attr-defined]
         calculated_ratio = wait_duration / prx_duration
 
         cpc_logger.debug(
@@ -531,7 +531,7 @@ def insert_dd_sequences(
                 # each PRX gate in the sequence must be implementable using a single drive channel segment
                 dd_segments = []
                 for theta, phi in prx_args:
-                    seg = _get_channel_segment(builder, prx(theta, phi), channel_name)
+                    seg = _get_channel_segment(builder, prx(theta, phi), channel_name)  # type: ignore[arg-type]
                     dd_segments.append(seg)
 
                 cpc_logger.debug(

{iqm_pulla-8.1.0 → iqm_pulla-8.3.0}/src/iqm/cpc/compiler/standard_stages.py

@@ -232,7 +232,7 @@ def _fix_implementation_and_locus(
         inst.locus = locus
         circuit_components.update(locus)
         if op.arity == 2:
-            circuit_component_pairs.add(locus)
+            circuit_component_pairs.add(locus)  # type: ignore[arg-type]
         circuit_gate_loci.setdefault(inst.name, {}).setdefault(inst.implementation, Counter()).update([locus])
 
     return CircuitMetrics(
@@ -274,7 +274,10 @@ def _build_readout_mappings(
 
     # find out which components have measurement data
     components_that_can_be_measured = frozenset(
-        q for impl_loci in builder.calibration["measure"].values() for locus in impl_loci for q in locus
+        q
+        for impl_loci in builder.calibration["measure"].values()
+        for locus in impl_loci
+        for q in locus  # type: ignore[union-attr]
     )
 
     # Figure out which qubits should always be measured on the QPU in the final measurement.
@@ -411,22 +414,22 @@ def _get_op_calibration_errors(calibration: OpCalibrationDataTree, ops: QuantumO
                 # since OILCalibrationData can have nested dicts, we do a recursive diff
                 error = diff_dicts(merge_dicts(default_cal_data, cal_data), impl.parameters, [])
                 if error is not None:
-                    errors[(op_name, impl_name, locus)] = error
+                    errors[(op_name, impl_name, locus)] = error  # type: ignore[index]
                     continue
 
                 n_components = len(locus)
                 arity = op.arity
                 if arity == 0:
                     if n_components != 1:
-                        errors[(op_name, impl_name, locus)] = (
+                        errors[(op_name, impl_name, locus)] = (  # type: ignore[index]
                             f"{op_name}.{impl_name} at {locus}: for zero-arity operations, "
                             "calibration data must be provided for single-component loci only"
                         )
                 elif n_components != arity:
-                    errors[(op_name, impl_name, locus)] = (
+                    errors[(op_name, impl_name, locus)] = (  # type: ignore[index]
                         f"{op_name}.{impl_name} at {locus}: locus must have {arity} component(s)"
                     )
-    return errors
+    return errors  # type: ignore[return-value]
 
 
 def merge_multiplexed_timeboxes(circuit_box: TimeBox) -> TimeBox:
@@ -469,7 +472,7 @@ def merge_multiplexed_timeboxes(circuit_box: TimeBox) -> TimeBox:
             if pending:
                 if disjoint_boxes(pending, gate_box):
                     # Pending box and new candidate have disjoint loci, merge is possible.
-                    pending = pending + gate_box.children[0]
+                    pending = pending + gate_box.children[0]  # type: ignore[assignment]
                     continue
                 # Pending box collides with the new candidate, so we must place it immediately and continue with
                 # the new candidate.
@@ -480,7 +483,7 @@ def merge_multiplexed_timeboxes(circuit_box: TimeBox) -> TimeBox:
             if pending and not disjoint_boxes(pending, gate_box):
                 placed_boxes.append(pending)
                 pending = None
-            placed_boxes.append(gate_box)
+            placed_boxes.append(gate_box)  # type: ignore[arg-type]
 
     if pending:
         placed_boxes.append(pending)
@@ -540,7 +543,7 @@ def map_components(
             _map_components_in_instructions(
                 component_mapping,
                 c.instructions,
-                device_components=device_components,
+                device_components=device_components,  # type: ignore[arg-type]
             )
         except CircuitError as exc:
             raise CircuitError(f"Circuit {idx}: {exc}") from exc

{iqm_pulla-8.1.0 → iqm_pulla-8.3.0}/src/iqm/cpc/compiler/station_settings.py

@@ -140,7 +140,7 @@ def find_observation(
     obs_value = calibration_set.get(observation_path)
     if obs_value is None and required:
         raise CalibrationError(f"Missing calibration observation: {observation_path}")
-    return obs_value
+    return obs_value  # type: ignore[return-value]
 
 
 # TODO until station type records (defining the instrumentation for each chip component) are available,
@@ -363,28 +363,28 @@ def build_station_settings(
     # Then add the required static settings for circuit execution.
     for qubit in circuit_qubits:
         apply_observations(qubit, _per_qubit)
-        _apply_static_settings(qubit, _per_qubit_static, root)
+        _apply_static_settings(qubit, _per_qubit_static, root)  # type: ignore[arg-type]
         if qubit in flux_pulsed_qubits:
             apply_observations(qubit, _per_flux_pulsed_qubit)
-            _apply_static_settings(qubit, _per_flux_pulsed_qubit_static, root)
+            _apply_static_settings(qubit, _per_flux_pulsed_qubit_static, root)  # type: ignore[arg-type]
 
     for coupler in circuit_couplers:
         apply_observations(coupler, _per_coupler)
-        _apply_static_settings(coupler, _per_coupler_static, root)
+        _apply_static_settings(coupler, _per_coupler_static, root)  # type: ignore[arg-type]
 
     for pl in measured_probe_lines:
         apply_observations(pl, _per_probe_line)
-        _apply_static_settings(pl, _per_probe_line_static, root)
+        _apply_static_settings(pl, _per_probe_line_static, root)  # type: ignore[arg-type]
 
     for qubit in boundary_qubits:
         apply_observations(qubit, _per_boundary_qubit)
         _apply_static_settings(qubit, _per_boundary_qubit_static, root)
         if qubit in flux_pulsed_qubits:
-            _apply_static_settings(qubit, _per_boundary_flux_pulsed_qubit_static, root)
+            _apply_static_settings(qubit, _per_boundary_flux_pulsed_qubit_static, root)  # type: ignore[arg-type]
 
     for coupler in boundary_couplers:
         apply_observations(coupler, _per_boundary_coupler)
-        _apply_static_settings(coupler, _per_boundary_coupler_static, root)
+        _apply_static_settings(coupler, _per_boundary_coupler_static, root)  # type: ignore[arg-type]
 
     # QPU-wide options
     apply_observations("", _per_qpu)

{iqm_pulla-8.1.0 → iqm_pulla-8.3.0}/src/iqm/cpc/interface/compiler.py

@@ -16,7 +16,7 @@
 from collections import Counter
 from dataclasses import dataclass, field
 from enum import StrEnum
-from typing import Final, TypeAlias
+from typing import TypeAlias
 
 from exa.common.data.setting_node import SettingNode
 from iqm.pulse.builder import CircuitOperation, Locus
@@ -44,10 +44,10 @@ class MeasurementMode(StrEnum):
     Measurement results which are not required by the circuits to be executed are discarded.
     """
 
-    CIRCUIT: Final[str] = "circuit"
+    CIRCUIT = "circuit"
     """In each circuit separately, measure only the components that have final measurement
     operations on them."""
-    ALL: Final[str] = "all"
+    ALL = "all"
     """Measure all the components on the QPU that have measurement data in the calset.
     This is typically how measurement is calibrated."""
 
@@ -55,9 +55,9 @@ class MeasurementMode(StrEnum):
 class HeraldingMode(StrEnum):
     """Heralding mode for circuit execution."""
 
-    NONE: Final[str] = "none"
+    NONE = "none"
     """Do not do any heralding."""
-    ZEROS: Final[str] = "zeros"
+    ZEROS = "zeros"
     """Perform a heralding measurement on all the components used in each circuit (if they have
     measurement data available in the calset), only retain shots where all the components
     are measured to be in the zero state."""
@@ -66,9 +66,9 @@ class HeraldingMode(StrEnum):
 class DDMode(StrEnum):
     """Dynamical Decoupling (DD) mode for circuit execution."""
 
-    DISABLED: Final[str] = "disabled"
+    DISABLED = "disabled"
     """Do not apply dynamical decoupling."""
-    ENABLED: Final[str] = "enabled"
+    ENABLED = "enabled"
     """Apply dynamical decoupling."""
 
 
@@ -127,7 +127,7 @@ class DDStrategy:
 class CircuitBoundaryMode(StrEnum):
     """Circuit boundary mode for circuit compilation."""
 
-    NEIGHBOUR: Final[str] = "neighbour"
+    NEIGHBOUR = "neighbour"
     """
     Circuit boundary consists of those QPU elements (qubits and couplers) that
     are adjacent to the qubits and couplers used by the circuit, but do not belong to them.
@@ -136,33 +136,33 @@ class CircuitBoundaryMode(StrEnum):
     * Boundary qubits are connected to a circuit qubit by any coupler, but are not circuit qubits themselves.
     * Boundary couplers are connected to at least one circuit qubit, but are not used in the circuit themselves.
     """
-    ALL: Final[str] = "all"
+    ALL = "all"
     """Circuit boundary consists of all the QPU elements that are not used in the circuit."""
 
 
 class MoveGateValidationMode(StrEnum):
     """MOVE gate validation mode for circuit compilation."""
 
-    STRICT: Final[str] = "strict"
+    STRICT = "strict"
     """Perform standard MOVE gate validation: MOVE(qubit, resonator) gates must only
     appear in sandwiches (pairs). Inside a sandwich there must be no gates acting on the
     MOVE qubit, and no other MOVE gates acting on the resonator."""
-    ALLOW_PRX: Final[str] = "allow_prx"
+    ALLOW_PRX = "allow_prx"
     """Allow PRX gates on the MOVE qubit inside MOVE sandwiches during validation."""
-    NONE: Final[str] = "none"
+    NONE = "none"
     """Do not perform any MOVE gate validation."""
 
 
 class MoveGateFrameTrackingMode(StrEnum):
     """MOVE gate frame tracking mode for circuit compilation."""
 
-    FULL: Final[str] = "full"
+    FULL = "full"
     """Perform complete MOVE gate frame tracking, applying both the explicit z rotations
     on the resonator and the dynamic phase correction due to qubit-resonator detuning to
     the qubit at the end of a MOVE sandwich."""
-    NO_DETUNING_CORRECTION: Final[str] = "no_detuning_correction"
+    NO_DETUNING_CORRECTION = "no_detuning_correction"
     """Do not apply the detuning correction at the end of a MOVE sandwich."""
-    NONE: Final[str] = "none"
+    NONE = "none"
     """Do not perform any MOVE gate frame tracking."""
 
 

{iqm_pulla-8.1.0 → iqm_pulla-8.3.0}/src/iqm/pulla/pulla.py

@@ -251,10 +251,10 @@ class Pulla:
                 sweep_data = self._station_control.get_sweep(job_id)
                 sc_result = StationControlResult(sweep_id=job_id, task_id=job_id, status=TaskStatus.PENDING)
 
-                if sweep_data.job_status <= JobExecutorStatus.EXECUTION_STARTED:
+                if sweep_data.job_status <= JobExecutorStatus.EXECUTION_STARTED:  # type: ignore[operator]
                     # Wait in the task queue while showing a progress bar
 
-                    interrupted = self._station_control._wait_job_completion(str(job_id), get_progress_bar_callback())
+                    interrupted = self._station_control._wait_job_completion(str(job_id), get_progress_bar_callback())  # type: ignore[attr-defined]
                     if interrupted:
                         raise KeyboardInterrupt
 
@@ -285,7 +285,7 @@ class Pulla:
                     )
                     sc_result.end_time = sweep_data.end_timestamp.isoformat() if sweep_data.end_timestamp else None
                     job = self._station_control.get_job(job_id)
-                    sc_result.message = job["job_error"]
+                    sc_result.message = job["job_error"]  # type: ignore[index]
                     logger.error("Submission failed! Error: %s", sc_result.message)
                     return sc_result
 
@@ -296,7 +296,7 @@ class Pulla:
                     )
                     sc_result.end_time = sweep_data.end_timestamp.isoformat() if sweep_data.end_timestamp else None
                     job = self._station_control.get_job(job_id)
-                    sc_result.message = job["job_error"]
+                    sc_result.message = job["job_error"]  # type: ignore[index]
                     logger.error("Submission was revoked!")
                     return sc_result
 

{iqm_pulla-8.1.0 → iqm_pulla-8.3.0}/src/iqm/pulla/quantum_architecture.py

@@ -40,8 +40,8 @@ def create_static_quantum_architecture(chip_topology: ChipTopology) -> StaticQua
     # The components in each connection are already sorted, now we sort the connections
     connectivity = sorted(unsorted_connections, key=sort_key)
     return StaticQuantumArchitecture(
-        qubits=chip_topology.qubits_sorted,
-        computational_resonators=chip_topology.computational_resonators_sorted,
+        qubits=chip_topology.qubits_sorted,  # type: ignore[arg-type]
+        computational_resonators=chip_topology.computational_resonators_sorted,  # type: ignore[arg-type]
         connectivity=connectivity,
     )
 

{iqm_pulla-8.1.0 → iqm_pulla-8.3.0}/src/iqm/pulla/utils.py

@@ -421,7 +421,7 @@ def calset_to_cal_data_tree(calibration_set: CalibrationSet) -> OpCalibrationDat
             locus = tuple(path[3].split(LOCUS_SEPARATOR))
             locus = () if locus == ("",) else locus
             # mypy likes this
-            set_path(tree.setdefault(path[1], {}).setdefault(path[2], {}).setdefault(locus, {}), path[4:], value)
+            set_path(tree.setdefault(path[1], {}).setdefault(path[2], {}).setdefault(locus, {}), path[4:], value)  # type: ignore[arg-type]
     return tree
 
 
@@ -483,7 +483,7 @@ def _update_channel_props_from_calibration(
                     f"No calibration value found for the center frequency or local oscillator frequency of {component}."
                 )
             channel_name = channels["readout"]
-            replacements[channel_name] = replace(channel_properties[channel_name], center_frequency=center_frequency)
+            replacements[channel_name] = replace(channel_properties[channel_name], center_frequency=center_frequency)  # type: ignore[call-arg]
 
     return channel_properties | replacements
 
@@ -518,8 +518,8 @@ def find_circuit_boundary(
         }
     elif mode == CircuitBoundaryMode.ALL:
         # maybe safer/better: all unused locus components/couplers are considered boundary
-        boundary_components = (device.qubits | device.computational_resonators) - circuit_components
-        boundary_couplers = device.couplers - circuit_couplers
+        boundary_components = (device.qubits | device.computational_resonators) - circuit_components  # type: ignore[assignment]
+        boundary_couplers = device.couplers - circuit_couplers  # type: ignore[assignment]
     else:
         raise UnknownCircuitExecutionOptionError(f"Unknown circuit boundary mode '{str(mode)}'")
     return boundary_components, boundary_couplers

{iqm_pulla-8.1.0 → iqm_pulla-8.3.0}/src/iqm/pulla/utils_qir.py

@@ -111,7 +111,7 @@ def _gate_inst_to_str(inst: Call) -> CircuitOperation | None:
     try:
         qir_logger.debug("Processing %s with args: %s", operation, inst.args)
         params = operation_handlers[operation](inst.args)
-        return CircuitOperation(**params)
+        return CircuitOperation(**params)  # type: ignore[arg-type]  # type: ignore[arg-type]  # type: ignore[arg-type]  # type: ignore[arg-type]
     except (IndexError, ValueError, AttributeError) as e:
         qir_logger.error("Error processing operation %s: %s", operation, e)
         raise ValueError(f"Error processing operation {operation}: {e}") from e

{iqm_pulla-8.1.0 → iqm_pulla-8.3.0}/src/iqm_pulla.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: iqm-pulla
-Version: 8.1.0
+Version: 8.3.0
 Summary: Client library for pulse-level access to an IQM quantum computer
 Author-email: IQM Finland Oy <developers@meetiqm.com>
 License:                                  Apache License

{iqm_pulla-8.1.0 → iqm_pulla-8.3.0}/src/iqm_pulla.egg-info/SOURCES.txt

@@ -55,6 +55,7 @@ src/iqm/pulla/__init__.py
 src/iqm/pulla/calibration.py
 src/iqm/pulla/interface.py
 src/iqm/pulla/pulla.py
+src/iqm/pulla/py.typed
 src/iqm/pulla/quantum_architecture.py
 src/iqm/pulla/utils.py
 src/iqm/pulla/utils_cirq.py

iqm_pulla-8.3.0/version.txt

@@ -0,0 +1 @@
+8.3.0

iqm_pulla-8.1.0/version.txt

@@ -1 +0,0 @@
-8.1.0