iqm-station-control-client 9.1.0__py3-none-any.whl → 9.3.0__py3-none-any.whl

iqm/station_control/client/iqm_server/iqm_server_client.py

@@ -96,7 +96,7 @@ class IqmServerClient(_StationControlClientBase):
     ):
         super().__init__(root_url, get_token_callback=get_token_callback, client_signature=client_signature)
         self._connection_params = parse_connection_params(root_url)
-        self._cached_resources = {}
+        self._cached_resources: dict[str, Any] = {}
         self._latest_submitted_sweep = None
         self._channel = grpc_channel or create_channel(self._connection_params, self._get_token_callback)
         self._current_qc = resolve_current_qc(self._channel, self._connection_params.quantum_computer)

iqm/station_control/client/iqm_server/proto/uuid_pb2.pyi

@@ -23,4 +23,4 @@ class Uuid(_message.Message):
     STR_FIELD_NUMBER: _ClassVar[int]
     raw: bytes
     str: str
-    def __init__(self, raw: _Optional[bytes] = ..., str: _Optional[str] = ...) -> None: ...
+    def __init__(self, raw: _Optional[bytes] = ..., str: _Optional[str] = ...) -> None: ...# type: ignore[valid-type]

iqm/station_control/client/list_models.py

@@ -64,12 +64,12 @@ class ListModel(RootModel):
     )
 
 
-DutList: TypeAlias = ListModel[list[DutData]]
-DutFieldDataList: TypeAlias = ListModel[list[DutFieldData]]
-ObservationDataList: TypeAlias = ListModel[list[ObservationData]]
-ObservationDefinitionList: TypeAlias = ListModel[list[ObservationDefinition]]
-ObservationLiteList: TypeAlias = ListModel[list[ObservationLite]]
-ObservationUpdateList: TypeAlias = ListModel[list[ObservationUpdate]]
-ObservationSetDataList: TypeAlias = ListModel[list[ObservationSetData]]
-SequenceMetadataDataList: TypeAlias = ListModel[list[SequenceMetadataData]]
-RunLiteList: TypeAlias = ListModel[list[RunLite]]
+DutList: TypeAlias = ListModel[list[DutData]]  # type: ignore[type-arg]
+DutFieldDataList: TypeAlias = ListModel[list[DutFieldData]]  # type: ignore[type-arg]
+ObservationDataList: TypeAlias = ListModel[list[ObservationData]]  # type: ignore[type-arg]
+ObservationDefinitionList: TypeAlias = ListModel[list[ObservationDefinition]]  # type: ignore[type-arg]
+ObservationLiteList: TypeAlias = ListModel[list[ObservationLite]]  # type: ignore[type-arg]
+ObservationUpdateList: TypeAlias = ListModel[list[ObservationUpdate]]  # type: ignore[type-arg]
+ObservationSetDataList: TypeAlias = ListModel[list[ObservationSetData]]  # type: ignore[type-arg]
+SequenceMetadataDataList: TypeAlias = ListModel[list[SequenceMetadataData]]  # type: ignore[type-arg]
+RunLiteList: TypeAlias = ListModel[list[RunLite]]  # type: ignore[type-arg]

iqm/station_control/client/station_control.py

@@ -251,7 +251,7 @@ class StationControlClient(_StationControlClientBase):
         super().__init__(
             root_url,
             get_token_callback=get_token_callback,
-            client_signature=client_signature,
+            client_signature=client_signature,  # type: ignore[arg-type]
             enable_opentelemetry=os.environ.get("JAEGER_OPENTELEMETRY_COLLECTOR_ENDPOINT", None) is not None,
         )
         # TODO SW-1387: Remove this when using v1 API, not needed
@@ -602,7 +602,7 @@ class StationControlClient(_StationControlClientBase):
 
     @staticmethod
     def _clean_query_parameters(model: Any, **kwargs) -> dict[str, Any]:
-        if issubclass(model, PydanticBase) and "invalid" in model.model_fields.keys() and "invalid" not in kwargs:
+        if issubclass(model, PydanticBase) and "invalid" in model.model_fields and "invalid" not in kwargs:
             # Get only valid items by default, "invalid=None" would return also invalid ones.
             # This default has to be set on the client side, server side uses default "None".
             kwargs["invalid"] = False
@@ -611,7 +611,7 @@ class StationControlClient(_StationControlClientBase):
     @staticmethod
     def _deserialize_response(
         response: requests.Response,
-        model_class: type[TypePydanticBase] | type[ListModel[list[TypePydanticBase]]],
+        model_class: type[TypePydanticBase | ListModel[list[TypePydanticBase]]],
         *,
         list_with_meta: bool = False,
     ) -> TypePydanticBase | ListWithMeta[TypePydanticBase]:

iqm/station_control/interface/models/dynamic_quantum_architecture.py

@@ -13,9 +13,10 @@
 # limitations under the License.
 """Dynamic quantum architecture (DQA) related interface models."""
 
+from typing import Any
 from uuid import UUID
 
-from pydantic import Field, StrictStr
+from pydantic import Field, StrictStr, field_validator
 
 from iqm.station_control.interface.pydantic_base import PydanticBase
 
@@ -59,6 +60,26 @@ class GateInfo(PydanticBase):
     )
     """Mapping of loci to implementation names that override ``default_implementation`` for those loci."""
 
+    @field_validator("override_default_implementation", mode="before")
+    @classmethod
+    def override_default_implementation_validator(cls, value: Any) -> dict[Locus, str]:
+        """Converts locus keys to tuples if they are encoded as strings."""
+        new_value = {}
+        if isinstance(value, dict):
+            for k, v in value.items():
+                if isinstance(k, tuple):
+                    new_value[k] = v
+                # When Pydantic serializes a dict with tuple keys into JSON, the keys are turned into
+                # comma-separated strings (because JSON only supports string keys). Here we convert
+                # them back into tuples.
+                elif isinstance(k, str):
+                    new_k = tuple(k.split(","))
+                    new_value[new_k] = v
+                else:
+                    raise ValueError("'override_default_implementation' keys must be strings or tuples.")
+            return new_value
+        raise ValueError("'override_default_implementation' must be a dict.")
+
 
 class DynamicQuantumArchitecture(PydanticBase):
     """The dynamic quantum architecture (DQA).

{iqm_station_control_client-9.1.0.dist-info → iqm_station_control_client-9.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: iqm-station-control-client
-Version: 9.1.0
+Version: 9.3.0
 Summary: Python client for communicating with Station Control Service
 Author-email: IQM Finland Oy <info@meetiqm.com>
 License: Apache License

{iqm_station_control_client-9.1.0.dist-info → iqm_station_control_client-9.3.0.dist-info}/RECORD

@@ -1,11 +1,11 @@
 iqm/station_control/client/__init__.py,sha256=1ND-AkIE9xLGIscH3WN44eyll9nlFhXeyCm-8EDFGQ4,942
-iqm/station_control/client/list_models.py,sha256=xnGk7Yzy-9bGBv_6eUzyai5hNJmG9ACUyu02ifSeSTE,2467
-iqm/station_control/client/station_control.py,sha256=4xUiI5Rw---TQDTWhN5FJb-nqhq6PE8q6DGSPfaH6l4,28154
+iqm/station_control/client/list_models.py,sha256=7JJyn5jCBOvMBJWdsVsPRLhtChMxqLa6O9hElpeXEt8,2701
+iqm/station_control/client/station_control.py,sha256=S-0YdGLKla0v98ntABbij7sXXxP-CwkynbzqQRf21s0,28167
 iqm/station_control/client/utils.py,sha256=-6K4KgOgA4iyUCqX-w26JvFxlwlGBehGj4tIWCEbn74,3360
 iqm/station_control/client/iqm_server/__init__.py,sha256=nLsRHN1rnOKXwuzaq_liUpAYV3sis5jkyHccSdacV7U,624
 iqm/station_control/client/iqm_server/error.py,sha256=a8l7UTnzfbD8KDHP-uOve77S6LR1ai9uM_J_xHbLs0Y,1175
 iqm/station_control/client/iqm_server/grpc_utils.py,sha256=67-uxYnNQ9NkC6t8V1_e9XFuIYv47K1I4A39MfHncVM,5825
-iqm/station_control/client/iqm_server/iqm_server_client.py,sha256=Ey478VqH6vZm7oVoBn7Ht6PzZHDJg0y6Smiqo9W1mdU,19601
+iqm/station_control/client/iqm_server/iqm_server_client.py,sha256=6kyk-cqyDX1Ihu0h3LRevVBADrNMOeBpH8sIyT0zT3A,19617
 iqm/station_control/client/iqm_server/proto/__init__.py,sha256=mOJQ_H-NEyJMffRaDSSZeXrScHaHaHEXULv-O_OJA3A,1345
 iqm/station_control/client/iqm_server/proto/calibration_pb2.py,sha256=gum0DGmqxhbfaar8SqahmSif1pB6hgo0pVcnoi3VMUo,3017
 iqm/station_control/client/iqm_server/proto/calibration_pb2.pyi,sha256=4lTHY_GhrsLIHqoGDkNLYu56QHzX_iHEbLaYq-HR1m8,2016
@@ -20,7 +20,7 @@ iqm/station_control/client/iqm_server/proto/qc_pb2.py,sha256=-ika0EK0l_I9V5-kcG2
 iqm/station_control/client/iqm_server/proto/qc_pb2.pyi,sha256=s4P3k2wA1XG1wRF_y9BQD7VQgDONQnhOzxItih8_d0I,2491
 iqm/station_control/client/iqm_server/proto/qc_pb2_grpc.py,sha256=6l6OECi0td_Ld7kd1MIrGjQmvs1F6Sd-vqeCfYeAS94,6958
 iqm/station_control/client/iqm_server/proto/uuid_pb2.py,sha256=H1O7qvtRg2zXqOVfWCjZH-2WjXdzyNjbSv2g0fKnfV0,1616
-iqm/station_control/client/iqm_server/proto/uuid_pb2.pyi,sha256=9LXcqNoQS1iapCsoMIZchKPP6-5jcbJqGLne9D4Fu5w,1029
+iqm/station_control/client/iqm_server/proto/uuid_pb2.pyi,sha256=C78m8tpY-sB9VbneOCpSCzrgjqFFu4a1eKBngQB0GuA,1055
 iqm/station_control/client/iqm_server/proto/uuid_pb2_grpc.py,sha256=SF40l84__r-OGGNYBru5ik9gih-XqeTq2iwM5gMN5Qc,726
 iqm/station_control/client/iqm_server/testing/__init__.py,sha256=wCNfJHIR_bqG3ZBlgm55v90Rih7VCpfctoIMfwRMgjk,567
 iqm/station_control/client/iqm_server/testing/iqm_server_mock.py,sha256=X_Chi8TKx95PiuhFfGnRu9LxeIpnKKynW_8tXwxFQD8,3340
@@ -39,7 +39,7 @@ iqm/station_control/interface/pydantic_base.py,sha256=pQCa-8SRgBKMSwG-KyXA1HcIxE
 iqm/station_control/interface/station_control.py,sha256=Uh9bDEZjc9M5KOYsGjPMYVhnYsfVQfjT05sUjUSoDXA,20303
 iqm/station_control/interface/models/__init__.py,sha256=Kg-XRLuU7G4Zjtu8LhYD-v3zkCYHzLFaNvC_nukezlQ,1968
 iqm/station_control/interface/models/dut.py,sha256=Hc_0XllXeIPGWhHsY7PC_jMpi7swpqo3jQQEQVnF3AM,1216
-iqm/station_control/interface/models/dynamic_quantum_architecture.py,sha256=5W8e5oRNxvXm2nI36NipDrErg2vHdpC8zzpeGyqSNDQ,3621
+iqm/station_control/interface/models/dynamic_quantum_architecture.py,sha256=3YlI9e1XxdbkBzlOdxmzYTccJ8AJRnRSIp-JoBf8Yyw,4698
 iqm/station_control/interface/models/jobs.py,sha256=4CVVGMsUOm_UO4w7UH-Shq7cbZe_5MkhGDpoBwHoHkc,5105
 iqm/station_control/interface/models/monitor.py,sha256=ItlgxtBup1hHg64uKeMKiFE7MaRRqSYdVRttsFD_XeU,1352
 iqm/station_control/interface/models/observation.py,sha256=FwhlI2xXZ_2-PA-LcDvscHuTbfdTWhw67zSNa9Buuvg,3261
@@ -49,8 +49,8 @@ iqm/station_control/interface/models/sequence.py,sha256=boWlMfP3woVgVObW3OaNbxsU
 iqm/station_control/interface/models/static_quantum_architecture.py,sha256=gsfJKlYsfZVEK3dqEKXkBSIHiY14DGwNbhPJdNHMtNM,1435
 iqm/station_control/interface/models/sweep.py,sha256=HFoFIrKhlYmHIBfGltY2O9_J28OvkkZILRbDHuqR0wc,2509
 iqm/station_control/interface/models/type_aliases.py,sha256=gEYJ8zOpV1m0NVyYUbAL43psp9Lxw_0t68mYlI65Sds,1262
-iqm_station_control_client-9.1.0.dist-info/LICENSE.txt,sha256=R6Q7eUrLyoCQgWYorQ8WJmVmWKYU3dxA3jYUp0wwQAw,11332
-iqm_station_control_client-9.1.0.dist-info/METADATA,sha256=dMBDpRAswdP4cKl7dZd81DCW4cQRKRtXyIeIoHa-9HM,14010
-iqm_station_control_client-9.1.0.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
-iqm_station_control_client-9.1.0.dist-info/top_level.txt,sha256=NB4XRfyDS6_wG9gMsyX-9LTU7kWnTQxNvkbzIxGv3-c,4
-iqm_station_control_client-9.1.0.dist-info/RECORD,,
+iqm_station_control_client-9.3.0.dist-info/LICENSE.txt,sha256=R6Q7eUrLyoCQgWYorQ8WJmVmWKYU3dxA3jYUp0wwQAw,11332
+iqm_station_control_client-9.3.0.dist-info/METADATA,sha256=Fsx6N7-afceIjDh8Fxzz7uZW7TDgcFIQr8__en45OlU,14010
+iqm_station_control_client-9.3.0.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
+iqm_station_control_client-9.3.0.dist-info/top_level.txt,sha256=NB4XRfyDS6_wG9gMsyX-9LTU7kWnTQxNvkbzIxGv3-c,4
+iqm_station_control_client-9.3.0.dist-info/RECORD,,