iqm-pulla 8.2.0__py3-none-any.whl → 9.0.0__py3-none-any.whl

iqm/cpc/compiler/compiler.py

@@ -49,7 +49,7 @@ from iqm.pulla.utils import (
 )
 from iqm.pulse.builder import ScheduleBuilder
 from iqm.pulse.gate_implementation import GateImplementation, Locus
-from iqm.pulse.gates import register_implementation
+from iqm.pulse.gates import register_implementation, register_operation
 
 # from iqm.pulse.gates.move import apply_move_gate_phase_corrections, validate_move_instructions
 from iqm.pulse.quantum_ops import QuantumOp
@@ -323,38 +323,36 @@ class Compiler:
         self,
         op_name: str,
         impl_name: str,
-        implementation: type[GateImplementation],
+        impl_class: type[GateImplementation],
         *,
         set_as_default: bool = False,
         overwrite: bool = False,
-        quantum_op_specs: QuantumOp | dict | None = None,
+        quantum_op: QuantumOp | None = None,
     ) -> None:
-        """Adds a new implementation of a quantum operation (gate).
+        """Adds a new implementation for a quantum operation (gate).
 
         Refreshes the compiler after adding a new implementation.
 
         Args:
-            op_name: The gate name for which to register a new implementation.
-            impl_name: The "human-readable" name with which the new gate implementation will be found e.g. in settings.
-            implementation: The python class of the new gate implementation to be added.
-            set_as_default: Whether to set the new implementation as the default implementation for the gate.
-            overwrite: If True, allows replacing any existing implementation of the same name.
-            quantum_op_specs: The quantum operation this gate represents. If a QuantumOp is given, it is used as is.
+            op_name: The name of the quantum operation for which to register a new implementation.
+            impl_name: The "human-readable" name with which the new implementation will be found e.g. in settings.
+            impl_class: The class of the new implementation to be added.
+            set_as_default: Whether to set the new implementation as the default implementation for the operation.
+            overwrite: If True, replaces any existing implementation of the same name for the operation.
+            quantum_op: The quantum operation this gate represents. If a QuantumOp is given, it is used as is.
                 If None is given and the same gate has been registered before, the previously registered properties are
-                used.
-                Otherwise, the given dict values are given to the constructor of
-                :class:`~iqm.pulse.quantum_ops.QuantumOp`.
-                For any missing constructor values, some defaults suitable for a 1-QB gate are used.
+                used. Existing operations cannot be replaced or modified.
 
         """
+        if quantum_op is not None:
+            register_operation(self.builder.op_table, quantum_op)
         register_implementation(
             operations=self.builder.op_table,
-            gate_name=op_name,
+            op_name=op_name,
             impl_name=impl_name,
-            impl_class=implementation,
+            impl_class=impl_class,
             set_as_default=set_as_default,
             overwrite=overwrite,
-            quantum_op_specs=quantum_op_specs,
         )
         self._refresh()
 

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/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/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/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/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/utils.py

@@ -86,7 +86,7 @@ def iqm_circuit_to_gate_implementation(circuit: CPC_Circuit, qubit_mapping: dict
         # TODO This method does not heed GateImplementation.locus, and will always apply the
         # gate on fixed qubits. It just pastes ``circuit`` contents, with the qubits mapped using
         # the likewise fixed mapping, into a TimeBox.
-        registered_gates = list({instr.name for instr in circuit.instructions})
+        registered_gates = tuple({instr.name for instr in circuit.instructions})
 
         def __call__(self):
             boxes = []
@@ -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/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.2.0.dist-info → iqm_pulla-9.0.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: iqm-pulla
-Version: 8.2.0
+Version: 9.0.0
 Summary: Client library for pulse-level access to an IQM quantum computer
 Author-email: IQM Finland Oy <developers@meetiqm.com>
 License: Apache License
@@ -218,27 +218,27 @@ License-File: LICENSE.txt
 License-File: AUTHORS.rst
 Requires-Dist: iqm-exa-common <27,>=26
 Requires-Dist: iqm-station-control-client <10,>=9
-Requires-Dist: iqm-pulse <10,>=9
+Requires-Dist: iqm-pulse <11,>=10
 Requires-Dist: iqm-data-definitions <3.0,>=2.13
 Requires-Dist: pylatexenc ==2.10
 Requires-Dist: pydantic <3.0,>=2.10.4
 Provides-Extra: notebook
 Requires-Dist: iqm-exa-common <27,>=26 ; extra == 'notebook'
 Requires-Dist: iqm-station-control-client <10,>=9 ; extra == 'notebook'
-Requires-Dist: iqm-pulse <10,>=9 ; extra == 'notebook'
+Requires-Dist: iqm-pulse <11,>=10 ; extra == 'notebook'
 Requires-Dist: notebook <7,>=6.4.11 ; extra == 'notebook'
 Requires-Dist: matplotlib <4,>=3.6.3 ; extra == 'notebook'
 Requires-Dist: nbclient ~=0.5.10 ; extra == 'notebook'
 Provides-Extra: qir
 Requires-Dist: iqm-exa-common <27,>=26 ; extra == 'qir'
 Requires-Dist: iqm-station-control-client <10,>=9 ; extra == 'qir'
-Requires-Dist: iqm-pulse <10,>=9 ; extra == 'qir'
+Requires-Dist: iqm-pulse <11,>=10 ; extra == 'qir'
 Requires-Dist: iqm-pyqir ==0.12.0 ; extra == 'qir'
 Requires-Dist: iqm-qiskit-qir ==0.8.0 ; extra == 'qir'
 Provides-Extra: qiskit
 Requires-Dist: iqm-exa-common <27,>=26 ; extra == 'qiskit'
 Requires-Dist: iqm-station-control-client <10,>=9 ; extra == 'qiskit'
-Requires-Dist: iqm-pulse <10,>=9 ; extra == 'qiskit'
+Requires-Dist: iqm-pulse <11,>=10 ; extra == 'qiskit'
 Requires-Dist: iqm-client <30,>=29 ; extra == 'qiskit'
 Requires-Dist: iqm-client[qiskit] <30,>=29 ; extra == 'qiskit'
 

{iqm_pulla-8.2.0.dist-info → iqm_pulla-9.0.0.dist-info}/RECORD

@@ -2,26 +2,27 @@ iqm/cpc/__init__.py,sha256=qw_SFgwld1YOYYyBE-NSticm60xx5aCAw5LJJzjou8Y,1745
 iqm/cpc/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 iqm/cpc/compiler/__init__.py,sha256=ieYyHzb2IyevDm5swU--rPMsh5GdUtLrw8773Cd0ens,678
 iqm/cpc/compiler/circuit_compilation_request_handler.py,sha256=Es3VoO1kOG2XHcJ5ntmT6-aMo7GiCUYkMJGILrUjkfg,4409
-iqm/cpc/compiler/compiler.py,sha256=9h1BrkGsO5VcyOCuXZd2dowHj-9fgK1Zemy5IH2YbaM,20512
-iqm/cpc/compiler/dd.py,sha256=Ks17njg6FBK7Yg6FBe9nOXCZ9QVFDYUGCGofbabHk0U,21618
+iqm/cpc/compiler/compiler.py,sha256=KsxRrpfPawZYqg3gvTYch58GXlSW8mr3p-11hejbZ_0,20383
+iqm/cpc/compiler/dd.py,sha256=j4onTod24RPxRwwsM3EMBBgX4hDO0k_nW9R762AocXo,21674
 iqm/cpc/compiler/errors.py,sha256=tz-8g1QtDvCAPmAjjCYK3FoULrazkpSTmXIvyqukaT4,1949
-iqm/cpc/compiler/standard_stages.py,sha256=eOPvdesEr-Oe469fil6ZyPUvzaWZHNjia5Vgkqr9jT8,33601
-iqm/cpc/compiler/station_settings.py,sha256=VuwD2TkraCXJql59nRE3RgiQXEmor8oCor1P1WC0rkk,16717
+iqm/cpc/compiler/standard_stages.py,sha256=W3CizQJDpxyz_4iHsQ-dxv7DxGxL_MMwOlIiiBjO6KY,33858
+iqm/cpc/compiler/station_settings.py,sha256=l4cJYcNtMar_BoChJaUbmVfpXv7liTCeLWpSv7fAipg,16903
 iqm/cpc/interface/__init__.py,sha256=mvhNx1I9K5Sg40CwPntWj35M3Bni__23PWEw_qYaXtQ,611
 iqm/cpc/interface/compiler.py,sha256=2x3ZhL4WcyjwlSoZ3KhlmRcLxboz_ptS0M_rdX1z0c8,9861
 iqm/pulla/__init__.py,sha256=fj5Qh8R9K-z6q5g9-CySBZsG8d33nU7hCHrqIIB8_-0,901
 iqm/pulla/calibration.py,sha256=c_SNlTxXC0y-ahDY7JY53N7SofU5WUWMrNBogtdfHl4,4523
 iqm/pulla/interface.py,sha256=yymQo4REHYcUOWgPxPYq9IixS9bBZ27LXLQgJhzET58,5400
-iqm/pulla/pulla.py,sha256=PTSU0ft0Q0gt4k6n_gj4sznxnT9cAgd37eGeDhu9S5A,14007
-iqm/pulla/quantum_architecture.py,sha256=T3rrv4sg1zD6bCY5Jb6Lp8yE3E0YVTh2nK0KZttjK6s,6208
-iqm/pulla/utils.py,sha256=nzUg0ASAaNHtfoY3TRMEzy90dLyoX3LEIKGKybS7Gm8,24623
+iqm/pulla/pulla.py,sha256=IpW6VLbpx7Qroy47sbybmNmv4jVr2fyd_zx-sdKMklA,14109
+iqm/pulla/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+iqm/pulla/quantum_architecture.py,sha256=mJH9e9fdkblg6LrgN7qd-Ui3Igawf-hGd2Z7ZclQg04,6260
+iqm/pulla/utils.py,sha256=TTAOFhNW59VFAjnu6ZicmHXvp-GI_f66oh6IC1zFQKE,24732
 iqm/pulla/utils_cirq.py,sha256=8SBy6w7cr4AmnCgKwh7dBWwBGfGKxnoEMv9-1yfKs0A,777
 iqm/pulla/utils_dd.py,sha256=SxYAuRBgvYELKjeXpFbP4mM0xCCivDk7WUHw7oEXfMo,1695
-iqm/pulla/utils_qir.py,sha256=s5igks5ctvmHiaFjT3eSAYyKWTbW-rPFE6NApN66CTE,10019
+iqm/pulla/utils_qir.py,sha256=VjhUFYm71LAdcx8OUgSaKwMg_zaFQfSEIuKDP0L_8fw,10123
 iqm/pulla/utils_qiskit.py,sha256=SyL9zz9xZ3mEi8t5uJVLxe3ipgNf4REZURMJQRr2E-E,10108
-iqm_pulla-8.2.0.dist-info/AUTHORS.rst,sha256=iCStz7WP5Jk7uMnn9jRA4ybS14X4yeUW2SsWE-OTaRk,328
-iqm_pulla-8.2.0.dist-info/LICENSE.txt,sha256=cCj_biRA4Q8A77vxR8AuvAf-DZ5G79yxR_3lYY6TrmA,11333
-iqm_pulla-8.2.0.dist-info/METADATA,sha256=VyQtf3k3VffAd6PMsEkEAsNZXA4qLjcDilBm8p4azd0,17691
-iqm_pulla-8.2.0.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
-iqm_pulla-8.2.0.dist-info/top_level.txt,sha256=NB4XRfyDS6_wG9gMsyX-9LTU7kWnTQxNvkbzIxGv3-c,4
-iqm_pulla-8.2.0.dist-info/RECORD,,
+iqm_pulla-9.0.0.dist-info/AUTHORS.rst,sha256=iCStz7WP5Jk7uMnn9jRA4ybS14X4yeUW2SsWE-OTaRk,328
+iqm_pulla-9.0.0.dist-info/LICENSE.txt,sha256=cCj_biRA4Q8A77vxR8AuvAf-DZ5G79yxR_3lYY6TrmA,11333
+iqm_pulla-9.0.0.dist-info/METADATA,sha256=wsZmPqqyjRETlQueBlE2PwvYWFORj1q65YEgjpGk77c,17695
+iqm_pulla-9.0.0.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
+iqm_pulla-9.0.0.dist-info/top_level.txt,sha256=NB4XRfyDS6_wG9gMsyX-9LTU7kWnTQxNvkbzIxGv3-c,4
+iqm_pulla-9.0.0.dist-info/RECORD,,