boulder-opal-scale-up-sdk 1.0.0__py3-none-any.whl → 1.0.2__py3-none-any.whl

boulderopalscaleupsdk/experiments/transmon_anharmonicity.py

@@ -0,0 +1,68 @@
+# Copyright 2025 Q-CTRL. All rights reserved.
+#
+# Licensed under the Q-CTRL Terms of service (the "License"). Unauthorized
+# copying or use of this file, via any medium, is strictly prohibited.
+# Proprietary and confidential. You may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+#    https://q-ctrl.com/terms
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS. See the
+# License for the specific language.
+
+from pydantic import PrivateAttr
+
+from .common import CWSIterable, Experiment, HypIterable, LinspaceIterable, RangeIterable
+
+DEFAULT_DURATION_NS = 2_000  # ns
+DEFAULT_RECYCLE_DELAY_NS = 1_000  # ns
+DEFAULT_SHOT_COUNT = 100
+DEFAULT_ANHARMONICITY_GUESS = -200e6  # Hz, typical anharmonicity for transmons
+
+
+class TransmonAnharmonicity(Experiment):
+    """
+    Parameters for running a Qubit Anharmonicity experiment.
+
+    Parameters
+    ----------
+    transmon : str
+        The transmon that is the target of the experiment.
+    frequencies : list[int] or LinspaceIterable or RangeIterable or CWSIterable
+                  or HypIterable or None, optional
+        The drive frequencies to be scanned in the experiment.
+        If None, a default scan will be used based on the transmon's frequency and anharmonicity.
+    spec_amplitude : float
+        The amplitude of the spectroscopy pulse, in volts.
+    spec_duration : int
+        The duration of the spectroscopy pulse, in nanoseconds.
+    anharmonicity_guess_hz : float
+        The guessed anharmonicity of the qubit, in Hz. This is used to determine the
+        range of frequencies to scan if `frequencies` is not provided.
+        Defaults to -200 MHz.
+    readout_amplitude : float
+        The amplitude of the readout pulse.
+    duration_ns : int, optional
+        The duration of the readout pulse, in nanoseconds.
+        Defaults to 2000 ns.
+    recycle_delay_ns : int, optional
+        The delay time between consecutive shots of the experiment, in nanoseconds.
+        Defaults to 1000 ns.
+    shot_count : int, optional
+        The number of shots to be taken in the experiment. Defaults to 100.
+    """
+
+    _experiment_name: str = PrivateAttr("transmon_anharmonicity")
+
+    transmon: str
+    frequencies: list[int] | LinspaceIterable | RangeIterable | CWSIterable | HypIterable | None = (
+        None
+    )
+    spec_amplitude: float
+    spec_duration: int
+    anharmonicity_guess_hz: float = DEFAULT_ANHARMONICITY_GUESS
+    readout_amplitude: float
+    duration_ns: int = DEFAULT_DURATION_NS
+    recycle_delay_ns: int = DEFAULT_RECYCLE_DELAY_NS
+    shot_count: int = DEFAULT_SHOT_COUNT

boulderopalscaleupsdk/experiments/transmon_spectroscopy.py

@@ -0,0 +1,69 @@
+# Copyright 2025 Q-CTRL. All rights reserved.
+#
+# Licensed under the Q-CTRL Terms of service (the "License"). Unauthorized
+# copying or use of this file, via any medium, is strictly prohibited.
+# Proprietary and confidential. You may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+#    https://q-ctrl.com/terms
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS. See the
+# License for the specific language.
+
+from pydantic import PrivateAttr
+
+from .common import (
+    CWSIterable,
+    Experiment,
+    HypIterable,
+    LinspaceIterable,
+    RangeIterable,
+)
+
+DEFAULT_DURATION_NS = 2000  # ns
+DEFAULT_RECYCLE_DELAY_NS = 1000  # ns
+DEFAULT_SHOT_COUNT = 100
+
+
+class TransmonSpectroscopy(Experiment):
+    """
+    Parameters for running a transmon spectroscopy experiment.
+
+    Parameters
+    ----------
+    transmon : str
+        The reference for the transmon to target in the experiment.
+    frequencies : list[int] or LinspaceIterable or RangeIterable or CWSIterable
+                  or HypIterable or None, optional
+        The frequencies at which to scan.
+        If None, frequencies around the readout frequency will be used.
+    spec_amplitude : float
+        The amplitude of the spectroscopy pulse.
+    spec_duration : int
+        The duration of the spectroscopy pulse, in nanoseconds.
+    readout_amplitude : float
+        The amplitude of the readout pulse.
+    duration_ns : int, optional
+        The duration of the readout pulse, in nanoseconds.
+        Defaults to 2000 ns.
+    recycle_delay_ns : int, optional
+        The delay time between consecutive shots of the experiment, in nanoseconds.
+        Defaults to 1000 ns.
+    shot_count : int, optional
+        The number of shots to be taken in the experiment.
+        Defaults to 100.
+    """
+
+    _experiment_name: str = PrivateAttr("transmon_spectroscopy")
+
+    transmon: str
+    frequencies: list[int] | LinspaceIterable | RangeIterable | CWSIterable | HypIterable | None = (
+        None
+    )
+    spec_amplitude: float
+    spec_duration: int
+    readout_amplitude: float
+    duration_ns: int = DEFAULT_DURATION_NS
+    recycle_delay_ns: int = DEFAULT_RECYCLE_DELAY_NS
+    shot_count: int = DEFAULT_SHOT_COUNT

boulderopalscaleupsdk/grpc_interceptors/auth.py

@@ -12,10 +12,13 @@
 # License for the specific language.
 
 import inspect
+from typing import TYPE_CHECKING
 
 import grpc
 import grpc.aio
-from qctrlclient import ApiKeyAuth
+
+if TYPE_CHECKING:
+    from qctrlclient import ApiKeyAuth  # pyright: ignore[reportMissingImports]
 
 
 class AuthInterceptor(
@@ -28,7 +31,7 @@ class AuthInterceptor(
     grpc.aio.StreamUnaryClientInterceptor,
     grpc.aio.StreamStreamClientInterceptor,
 ):
-    def __init__(self, auth: ApiKeyAuth):
+    def __init__(self, auth: "ApiKeyAuth"):
         self.auth = auth
 
     def _add_auth_metadata(self, client_call_details):

boulderopalscaleupsdk/plotting/__init__.py

@@ -14,11 +14,29 @@
 """Plotting library."""
 
 __all__ = [
+    "HeatmapData",
+    "HistogramData",
+    "HistogramPlot",
+    "LinePlot",
     "Marker",
     "Plot",
+    "PlotConfig",
     "PlotData1D",
-    "PlotData2D",
+    "PlotReport",
+    "Ticks",
     "VLine",
 ]
 
-from .dtypes import Marker, Plot, PlotData1D, PlotData2D, VLine
+from .dtypes import (
+    HeatmapData,
+    HistogramData,
+    HistogramPlot,
+    LinePlot,
+    Marker,
+    Plot,
+    PlotConfig,
+    PlotData1D,
+    PlotReport,
+    Ticks,
+    VLine,
+)

boulderopalscaleupsdk/plotting/dtypes.py

@@ -19,9 +19,9 @@ from pydantic import (
     BaseModel,
     BeforeValidator,
     ConfigDict,
+    Field,
     PlainSerializer,
     ValidationError,
-    model_validator,
 )
 from pydantic.dataclasses import dataclass
 
@@ -35,17 +35,13 @@ def _array_validator(value: Any) -> np.ndarray:
         array = np.asarray(value, order="C")
 
     if array.dtype == np.dtypes.ObjectDType:
-        raise ValidationError("Invalid array.")
+        raise ValidationError("Invalid array type.")
 
     return array
 
 
 def _array_serializer(array: np.ndarray) -> dict[str, Any]:
-    return {
-        "data": base64.b64encode(array),
-        "dtype": str(array.dtype),
-        "shape": array.shape,
-    }
+    return {"data": base64.b64encode(array), "dtype": str(array.dtype), "shape": array.shape}
 
 
 _SerializableArray = Annotated[
@@ -55,29 +51,51 @@ _SerializableArray = Annotated[
 ]
 
 
+@dataclass
+class PlotReport:
+    text: str
+    title: str = "Report"
+
+
+@dataclass
+class Ticks:
+    values: list[float]
+    labels: list[str]
+
+    def __post_init__(self):
+        if len(self.values) != len(self.labels):
+            raise ValueError("Ticks values and labels must have the same length.")
+
+
+class PlotConfig(BaseModel):
+    model_config = ConfigDict(extra="forbid")
+
+    title: str
+
+    x_label: str
+    y_label: str
+
+    x_bounds: tuple[float, float] | None = None
+    y_bounds: tuple[float, float] | None = None
+
+    x_ticks: Ticks | None = None
+    y_ticks: Ticks | None = None
+
+    reverse_yaxis: bool = False
+
+    report: PlotReport | None = None
+
+
 @dataclass(config=ConfigDict(arbitrary_types_allowed=True))
 class PlotData1D:
-    """
-    A class to represent 1D plot data with optional error bars.
-
-    Attributes
-    ----------
-    x : np.ndarray
-        The x-coordinates of the data points.
-    y : np.ndarray
-        The y-coordinates of the data points.
-    x_error : np.ndarray or None, optional
-        The errors in the x-coordinates. Defaults to None.
-    y_error : np.ndarray or None, optional
-        The errors in the y-coordinates. Defaults to None.
-    label : str or None, optional
-        The label for the data to display in the legend. Defaults to None.
-    """
-
     x: _SerializableArray
     y: _SerializableArray
+
+    style: Literal["dash", "solid", "scatter"]
+
     x_error: _SerializableArray | None = None
     y_error: _SerializableArray | None = None
+
     label: str | None = None
 
     def __post_init__(self):
@@ -87,7 +105,6 @@ class PlotData1D:
             raise ValueError("y must be 1D.")
         if len(self.x) != len(self.y):
             raise ValueError("The length of x and y must match.")
-
         if self.x_error is not None and self.x_error.shape != self.x.shape:
             raise ValueError("The shapes of x and x_error must match.")
         if self.y_error is not None and self.y_error.shape != self.y.shape:
@@ -95,27 +112,16 @@ class PlotData1D:
 
 
 @dataclass(config=ConfigDict(arbitrary_types_allowed=True))
-class PlotData2D:
-    """
-    A class to represent 2D plot data.
-
-    Attributes
-    ----------
-    x : np.ndarray
-        The x-coordinates of the data points.
-    y : np.ndarray
-        The y-coordinates of the data points.
-    z : np.ndarray
-        The z-values corresponding to each (x, y) pair.
-    label : str or None, optional
-        The label for the data to display in the legend. Defaults to None.
-    """
-
+class HeatmapData:
     x: _SerializableArray
     y: _SerializableArray
     z: _SerializableArray
+
     label: str | None = None
 
+    # Whether to display the heatmap values as text.
+    heatmap_text: bool = False
+
     def __post_init__(self):
         if self.x.ndim != 1:
             raise ValueError("x must be 1D.")
@@ -143,79 +149,37 @@ class VLine:
     color: str | None = None
 
 
-class Plot(BaseModel):
-    """
-    Data to plot the results of an experiment.
-
-    Parameters
-    ----------
-    heatmap : PlotData2D or None, optional
-        The 2D experimental data.
-        If provided, it's plotted as a heatmap.
-    heatmap_text : bool, optional
-        If True, the heatmap displays the values as text.
-        Defaults to False.
-    points : PlotData1D or None, optional
-        The 1D experimental data.
-        If provided, it's plotted as a scatter plot.
-    best_fit : PlotData1D or None, optional
-        The best fit on the experimental data.
-        If provided, it's plotted as a line plot.
-    reference_fit : PlotData1D or None, optional
-        A reference fit on the experimental data.
-        If provided, it's plotted as a line plot.
-    markers : list[Markers] or None, optional
-        Markers to add to the plot.
-    vlines : list[VLine] or None, optional
-        Vertical lines to add to the plot.
-    title : str or None, optional
-        The title of the plot.
-    xticks : list[float] or None, optional
-        The values at which x-ticks are placed.
-        Must be specified alongside xticklabels. Defaults to None.
-    xticklabels : list[str] or None, optional
-        The labels for the x-ticks.
-        Must be specified alongside xticks. Defaults to None.
-    yticks : list[float] or None, optional
-        The values at which y-ticks are placed.
-        Must be specified alongside yticklabels. Defaults to None.
-    yticklabels : list[str] or None, optional
-        The labels for the y-ticks.
-        Must be specified alongside xticks. Defaults to None.
-    x_label : str, optional
-        The label for the x-axis. Defaults to "X-axis".
-    y_label : str, optional
-        The label for the y-axis. Defaults to "Y-axis".
-    reverse_yaxis : bool, optional
-        If True, the y-axis is reversed. Defaults to False.
-    fit_report : str or None, optional
-        An optional report for the fit.
-    """
-
-    heatmap: PlotData2D | None = None
-    heatmap_text: bool = False
-    points: PlotData1D | None = None
-    best_fit: PlotData1D | None = None
-    reference_fit: PlotData1D | None = None
-    markers: list[Marker] | None = None
-    vlines: list[VLine] | None = None
-    title: str | None = None
-    xticks: list[float] | None = None
-    xticklabels: list[str] | None = None
-    yticks: list[float] | None = None
-    yticklabels: list[str] | None = None
-    x_label: str = "X-axis"
-    y_label: str = "Y-axis"
-    reverse_yaxis: bool | None = False
-    fit_report: str | None = None
-
-    @model_validator(mode="after")
-    def validate_ticks(self) -> "Plot":
-        # Check ticks and ticklabels consistency.
-        if not ((self.xticks is not None) ^ (self.xticklabels is None)):
-            raise ValueError("Both xticks and xticklabels must be provided together.")
-
-        if not ((self.yticks is not None) ^ (self.yticklabels is None)):
-            raise ValueError("Both yticks and yticklabels must be provided together.")
-
-        return self
+class LinePlot(BaseModel):
+    model_config = ConfigDict(extra="forbid")
+
+    plot_type: Literal["line_plot"] = "line_plot"
+
+    config: PlotConfig
+    heatmap: HeatmapData | None = None
+    lines: Annotated[list[PlotData1D], Field(min_length=1)]
+    markers: list[Marker] = []
+    vlines: list[VLine] = []
+
+
+@dataclass(config=ConfigDict(arbitrary_types_allowed=True))
+class HistogramData:
+    data: _SerializableArray
+    label: str | None = None
+    opacity: float = 0.7
+
+    def __post_init__(self):
+        if self.data.ndim != 1:
+            raise ValueError("data must be 1D.")
+
+
+class HistogramPlot(BaseModel):
+    model_config = ConfigDict(extra="forbid")
+
+    plot_type: Literal["histogram"] = "histogram"
+
+    config: PlotConfig
+    histograms: list[HistogramData]
+    vlines: list[VLine] = []
+
+
+Plot = LinePlot | HistogramPlot

boulderopalscaleupsdk/protobuf/v1/agent_pb2.py

@@ -17,7 +17,7 @@ from google.protobuf import empty_pb2 as google_dot_protobuf_dot_empty__pb2
 from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n-boulderopalscaleupsdk/protobuf/v1/agent.proto\x12!boulderopalscaleupsdk.protobuf.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a\x1cgoogle/protobuf/struct.proto\"H\n\x14RunQuaProgramRequest\x12\x18\n\x07program\x18\x01 \x01(\tR\x07program\x12\x16\n\x06\x63onfig\x18\x02 \x01(\tR\x06\x63onfig\"K\n\x15RunQuaProgramResponse\x12\x32\n\x08raw_data\x18\x01 \x01(\x0b\x32\x17.google.protobuf.StructR\x07rawData\"_\n)RunQuantumMachinesMixerCalibrationRequest\x12\x1a\n\x08\x65lements\x18\x01 \x03(\tR\x08\x65lements\x12\x16\n\x06\x63onfig\x18\x02 \x01(\tR\x06\x63onfig\"k\n*RunQuantumMachinesMixerCalibrationResponse\x12\x18\n\x07success\x18\x01 \x01(\x08R\x07success\x12\x19\n\x05\x65rror\x18\x02 \x01(\tH\x00R\x05\x65rror\x88\x01\x01\x42\x08\n\x06_error\"G\n\x15\x44isplayResultsRequest\x12\x18\n\x07message\x18\x01 \x01(\tR\x07message\x12\x14\n\x05plots\x18\x02 \x03(\tR\x05plots\"F\n\x16\x44isplayResultsResponse\x12,\n\x05\x65mpty\x18\x01 \x01(\x0b\x32\x16.google.protobuf.EmptyR\x05\x65mpty2\xdd\x04\n\x0c\x41gentService\x12\xa8\x01\n\rRunQuaProgram\x12\x37.boulderopalscaleupsdk.protobuf.v1.RunQuaProgramRequest\x1a\x38.boulderopalscaleupsdk.protobuf.v1.RunQuaProgramResponse\"$\x82\xd3\xe4\x93\x02\x1e\"\x19/v1/agent/run_qua_program:\x01*\x12\xf3\x01\n\"RunQuantumMachinesMixerCalibration\x12L.boulderopalscaleupsdk.protobuf.v1.RunQuantumMachinesMixerCalibrationRequest\x1aM.boulderopalscaleupsdk.protobuf.v1.RunQuantumMachinesMixerCalibrationResponse\"0\x82\xd3\xe4\x93\x02*\"%/v1/agent/run_qua_calibration_program:\x01*\x12\xab\x01\n\x0e\x44isplayResults\x12\x38.boulderopalscaleupsdk.protobuf.v1.DisplayResultsRequest\x1a\x39.boulderopalscaleupsdk.protobuf.v1.DisplayResultsResponse\"$\x82\xd3\xe4\x93\x02\x1e\"\x19/v1/agent/display_results:\x01*B\xa9\x02\n%com.boulderopalscaleupsdk.protobuf.v1B\nAgentProtoP\x01ZNgithub.com/qctrl/boulder-opal-scale-up/proto/boulderopalscaleupsdk/protobuf/v1\xa2\x02\x03\x42PX\xaa\x02!Boulderopalscaleupsdk.Protobuf.V1\xca\x02!Boulderopalscaleupsdk\\Protobuf\\V1\xe2\x02-Boulderopalscaleupsdk\\Protobuf\\V1\\GPBMetadata\xea\x02#Boulderopalscaleupsdk::Protobuf::V1b\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n-boulderopalscaleupsdk/protobuf/v1/agent.proto\x12!boulderopalscaleupsdk.protobuf.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a\x1cgoogle/protobuf/struct.proto\"\x85\x01\n\x11RunProgramRequest\x12\'\n\x0f\x63ontroller_type\x18\x01 \x01(\tR\x0e\x63ontrollerType\x12\x18\n\x07program\x18\x02 \x01(\tR\x07program\x12-\n\x12\x63\x61librate_elements\x18\x03 \x03(\tR\x11\x63\x61librateElements\"H\n\x12RunProgramResponse\x12\x32\n\x08raw_data\x18\x01 \x01(\x0b\x32\x17.google.protobuf.StructR\x07rawData\"_\n)RunQuantumMachinesMixerCalibrationRequest\x12\x1a\n\x08\x65lements\x18\x01 \x03(\tR\x08\x65lements\x12\x16\n\x06\x63onfig\x18\x02 \x01(\tR\x06\x63onfig\"k\n*RunQuantumMachinesMixerCalibrationResponse\x12\x18\n\x07success\x18\x01 \x01(\x08R\x07success\x12\x19\n\x05\x65rror\x18\x02 \x01(\tH\x00R\x05\x65rror\x88\x01\x01\x42\x08\n\x06_error\"G\n\x15\x44isplayResultsRequest\x12\x18\n\x07message\x18\x01 \x01(\tR\x07message\x12\x14\n\x05plots\x18\x02 \x03(\tR\x05plots\"F\n\x16\x44isplayResultsResponse\x12,\n\x05\x65mpty\x18\x01 \x01(\x0b\x32\x16.google.protobuf.EmptyR\x05\x65mpty2\xd0\x04\n\x0c\x41gentService\x12\x9b\x01\n\nRunProgram\x12\x34.boulderopalscaleupsdk.protobuf.v1.RunProgramRequest\x1a\x35.boulderopalscaleupsdk.protobuf.v1.RunProgramResponse\" \x82\xd3\xe4\x93\x02\x1a\"\x15/v1/agent/run_program:\x01*\x12\xf3\x01\n\"RunQuantumMachinesMixerCalibration\x12L.boulderopalscaleupsdk.protobuf.v1.RunQuantumMachinesMixerCalibrationRequest\x1aM.boulderopalscaleupsdk.protobuf.v1.RunQuantumMachinesMixerCalibrationResponse\"0\x82\xd3\xe4\x93\x02*\"%/v1/agent/run_qua_calibration_program:\x01*\x12\xab\x01\n\x0e\x44isplayResults\x12\x38.boulderopalscaleupsdk.protobuf.v1.DisplayResultsRequest\x1a\x39.boulderopalscaleupsdk.protobuf.v1.DisplayResultsResponse\"$\x82\xd3\xe4\x93\x02\x1e\"\x19/v1/agent/display_results:\x01*B\xa9\x02\n%com.boulderopalscaleupsdk.protobuf.v1B\nAgentProtoP\x01ZNgithub.com/qctrl/boulder-opal-scale-up/proto/boulderopalscaleupsdk/protobuf/v1\xa2\x02\x03\x42PX\xaa\x02!Boulderopalscaleupsdk.Protobuf.V1\xca\x02!Boulderopalscaleupsdk\\Protobuf\\V1\xe2\x02-Boulderopalscaleupsdk\\Protobuf\\V1\\GPBMetadata\xea\x02#Boulderopalscaleupsdk::Protobuf::V1b\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -25,24 +25,24 @@ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'boulderopalscaleupsdk.proto
 if not _descriptor._USE_C_DESCRIPTORS:
   _globals['DESCRIPTOR']._loaded_options = None
   _globals['DESCRIPTOR']._serialized_options = b'\n%com.boulderopalscaleupsdk.protobuf.v1B\nAgentProtoP\001ZNgithub.com/qctrl/boulder-opal-scale-up/proto/boulderopalscaleupsdk/protobuf/v1\242\002\003BPX\252\002!Boulderopalscaleupsdk.Protobuf.V1\312\002!Boulderopalscaleupsdk\\Protobuf\\V1\342\002-Boulderopalscaleupsdk\\Protobuf\\V1\\GPBMetadata\352\002#Boulderopalscaleupsdk::Protobuf::V1'
-  _globals['_AGENTSERVICE'].methods_by_name['RunQuaProgram']._loaded_options = None
-  _globals['_AGENTSERVICE'].methods_by_name['RunQuaProgram']._serialized_options = b'\202\323\344\223\002\036\"\031/v1/agent/run_qua_program:\001*'
+  _globals['_AGENTSERVICE'].methods_by_name['RunProgram']._loaded_options = None
+  _globals['_AGENTSERVICE'].methods_by_name['RunProgram']._serialized_options = b'\202\323\344\223\002\032\"\025/v1/agent/run_program:\001*'
   _globals['_AGENTSERVICE'].methods_by_name['RunQuantumMachinesMixerCalibration']._loaded_options = None
   _globals['_AGENTSERVICE'].methods_by_name['RunQuantumMachinesMixerCalibration']._serialized_options = b'\202\323\344\223\002*\"%/v1/agent/run_qua_calibration_program:\001*'
   _globals['_AGENTSERVICE'].methods_by_name['DisplayResults']._loaded_options = None
   _globals['_AGENTSERVICE'].methods_by_name['DisplayResults']._serialized_options = b'\202\323\344\223\002\036\"\031/v1/agent/display_results:\001*'
-  _globals['_RUNQUAPROGRAMREQUEST']._serialized_start=173
-  _globals['_RUNQUAPROGRAMREQUEST']._serialized_end=245
-  _globals['_RUNQUAPROGRAMRESPONSE']._serialized_start=247
-  _globals['_RUNQUAPROGRAMRESPONSE']._serialized_end=322
-  _globals['_RUNQUANTUMMACHINESMIXERCALIBRATIONREQUEST']._serialized_start=324
-  _globals['_RUNQUANTUMMACHINESMIXERCALIBRATIONREQUEST']._serialized_end=419
-  _globals['_RUNQUANTUMMACHINESMIXERCALIBRATIONRESPONSE']._serialized_start=421
-  _globals['_RUNQUANTUMMACHINESMIXERCALIBRATIONRESPONSE']._serialized_end=528
-  _globals['_DISPLAYRESULTSREQUEST']._serialized_start=530
-  _globals['_DISPLAYRESULTSREQUEST']._serialized_end=601
-  _globals['_DISPLAYRESULTSRESPONSE']._serialized_start=603
-  _globals['_DISPLAYRESULTSRESPONSE']._serialized_end=673
-  _globals['_AGENTSERVICE']._serialized_start=676
-  _globals['_AGENTSERVICE']._serialized_end=1281
+  _globals['_RUNPROGRAMREQUEST']._serialized_start=174
+  _globals['_RUNPROGRAMREQUEST']._serialized_end=307
+  _globals['_RUNPROGRAMRESPONSE']._serialized_start=309
+  _globals['_RUNPROGRAMRESPONSE']._serialized_end=381
+  _globals['_RUNQUANTUMMACHINESMIXERCALIBRATIONREQUEST']._serialized_start=383
+  _globals['_RUNQUANTUMMACHINESMIXERCALIBRATIONREQUEST']._serialized_end=478
+  _globals['_RUNQUANTUMMACHINESMIXERCALIBRATIONRESPONSE']._serialized_start=480
+  _globals['_RUNQUANTUMMACHINESMIXERCALIBRATIONRESPONSE']._serialized_end=587
+  _globals['_DISPLAYRESULTSREQUEST']._serialized_start=589
+  _globals['_DISPLAYRESULTSREQUEST']._serialized_end=660
+  _globals['_DISPLAYRESULTSRESPONSE']._serialized_start=662
+  _globals['_DISPLAYRESULTSRESPONSE']._serialized_end=732
+  _globals['_AGENTSERVICE']._serialized_start=735
+  _globals['_AGENTSERVICE']._serialized_end=1327
 # @@protoc_insertion_point(module_scope)

boulderopalscaleupsdk/protobuf/v1/agent_pb2.pyi

@@ -8,15 +8,17 @@ from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Map
 
 DESCRIPTOR: _descriptor.FileDescriptor
 
-class RunQuaProgramRequest(_message.Message):
-    __slots__ = ("program", "config")
+class RunProgramRequest(_message.Message):
+    __slots__ = ("controller_type", "program", "calibrate_elements")
+    CONTROLLER_TYPE_FIELD_NUMBER: _ClassVar[int]
     PROGRAM_FIELD_NUMBER: _ClassVar[int]
-    CONFIG_FIELD_NUMBER: _ClassVar[int]
+    CALIBRATE_ELEMENTS_FIELD_NUMBER: _ClassVar[int]
+    controller_type: str
     program: str
-    config: str
-    def __init__(self, program: _Optional[str] = ..., config: _Optional[str] = ...) -> None: ...
+    calibrate_elements: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, controller_type: _Optional[str] = ..., program: _Optional[str] = ..., calibrate_elements: _Optional[_Iterable[str]] = ...) -> None: ...
 
-class RunQuaProgramResponse(_message.Message):
+class RunProgramResponse(_message.Message):
     __slots__ = ("raw_data",)
     RAW_DATA_FIELD_NUMBER: _ClassVar[int]
     raw_data: _struct_pb2.Struct

boulderopalscaleupsdk/protobuf/v1/agent_pb2_grpc.py

@@ -15,10 +15,10 @@ class AgentServiceStub(object):
         Args:
             channel: A grpc.Channel.
         """
-        self.RunQuaProgram = channel.unary_unary(
-                '/boulderopalscaleupsdk.protobuf.v1.AgentService/RunQuaProgram',
-                request_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_agent__pb2.RunQuaProgramRequest.SerializeToString,
-                response_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_agent__pb2.RunQuaProgramResponse.FromString,
+        self.RunProgram = channel.unary_unary(
+                '/boulderopalscaleupsdk.protobuf.v1.AgentService/RunProgram',
+                request_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_agent__pb2.RunProgramRequest.SerializeToString,
+                response_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_agent__pb2.RunProgramResponse.FromString,
                 )
         self.RunQuantumMachinesMixerCalibration = channel.unary_unary(
                 '/boulderopalscaleupsdk.protobuf.v1.AgentService/RunQuantumMachinesMixerCalibration',
@@ -36,7 +36,7 @@ class AgentServiceServicer(object):
     """Defines the Agent service running locally.
     """
 
-    def RunQuaProgram(self, request, context):
+    def RunProgram(self, request, context):
         """Run a program on physical hardware.
         """
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
@@ -60,10 +60,10 @@ class AgentServiceServicer(object):
 
 def add_AgentServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
-            'RunQuaProgram': grpc.unary_unary_rpc_method_handler(
-                    servicer.RunQuaProgram,
-                    request_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_agent__pb2.RunQuaProgramRequest.FromString,
-                    response_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_agent__pb2.RunQuaProgramResponse.SerializeToString,
+            'RunProgram': grpc.unary_unary_rpc_method_handler(
+                    servicer.RunProgram,
+                    request_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_agent__pb2.RunProgramRequest.FromString,
+                    response_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_agent__pb2.RunProgramResponse.SerializeToString,
             ),
             'RunQuantumMachinesMixerCalibration': grpc.unary_unary_rpc_method_handler(
                     servicer.RunQuantumMachinesMixerCalibration,
@@ -87,7 +87,7 @@ class AgentService(object):
     """
 
     @staticmethod
-    def RunQuaProgram(request,
+    def RunProgram(request,
             target,
             options=(),
             channel_credentials=None,
@@ -97,9 +97,9 @@ class AgentService(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/boulderopalscaleupsdk.protobuf.v1.AgentService/RunQuaProgram',
-            boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_agent__pb2.RunQuaProgramRequest.SerializeToString,
-            boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_agent__pb2.RunQuaProgramResponse.FromString,
+        return grpc.experimental.unary_unary(request, target, '/boulderopalscaleupsdk.protobuf.v1.AgentService/RunProgram',
+            boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_agent__pb2.RunProgramRequest.SerializeToString,
+            boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_agent__pb2.RunProgramResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)