boulder-opal-scale-up-sdk 1.0.5__py3-none-any.whl → 1.0.6__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (29) hide show
  1. {boulder_opal_scale_up_sdk-1.0.5.dist-info → boulder_opal_scale_up_sdk-1.0.6.dist-info}/METADATA +1 -1
  2. {boulder_opal_scale_up_sdk-1.0.5.dist-info → boulder_opal_scale_up_sdk-1.0.6.dist-info}/RECORD +27 -21
  3. boulderopalscaleupsdk/agent/worker.py +15 -2
  4. boulderopalscaleupsdk/common/dtypes.py +21 -1
  5. boulderopalscaleupsdk/{stubs/__init__.py → constants.py} +3 -0
  6. boulderopalscaleupsdk/device/controller/quantum_machines.py +86 -17
  7. boulderopalscaleupsdk/device/processor/common.py +3 -3
  8. boulderopalscaleupsdk/errors.py +21 -0
  9. boulderopalscaleupsdk/experiments/__init__.py +8 -2
  10. boulderopalscaleupsdk/experiments/cz_spectroscopy_by_bias.py +84 -0
  11. boulderopalscaleupsdk/experiments/ramsey_ef.py +62 -0
  12. boulderopalscaleupsdk/experiments/{readout_classifier_calibration.py → readout_classifier.py} +7 -3
  13. boulderopalscaleupsdk/experiments/readout_optimization.py +57 -0
  14. boulderopalscaleupsdk/experiments/resonator_spectroscopy_by_bias.py +1 -3
  15. boulderopalscaleupsdk/experiments/transmon_anharmonicity.py +0 -2
  16. boulderopalscaleupsdk/grpc_interceptors/error.py +318 -0
  17. boulderopalscaleupsdk/plotting/dtypes.py +5 -5
  18. boulderopalscaleupsdk/protobuf/v1/device_pb2.py +57 -49
  19. boulderopalscaleupsdk/protobuf/v1/device_pb2.pyi +67 -43
  20. boulderopalscaleupsdk/protobuf/v1/device_pb2_grpc.py +100 -66
  21. boulderopalscaleupsdk/protobuf/v1/job_pb2.py +47 -0
  22. boulderopalscaleupsdk/protobuf/v1/job_pb2.pyi +54 -0
  23. boulderopalscaleupsdk/protobuf/v1/job_pb2_grpc.py +138 -0
  24. boulderopalscaleupsdk/third_party/quantum_machines/__init__.py +1 -1
  25. boulderopalscaleupsdk/third_party/quantum_machines/config.py +51 -48
  26. boulderopalscaleupsdk/stubs/dtypes.py +0 -47
  27. boulderopalscaleupsdk/stubs/maps.py +0 -18
  28. {boulder_opal_scale_up_sdk-1.0.5.dist-info → boulder_opal_scale_up_sdk-1.0.6.dist-info}/LICENSE +0 -0
  29. {boulder_opal_scale_up_sdk-1.0.5.dist-info → boulder_opal_scale_up_sdk-1.0.6.dist-info}/WHEEL +0 -0
@@ -1,57 +1,51 @@
1
1
  from google.api import annotations_pb2 as _annotations_pb2
2
2
  from google.protobuf import struct_pb2 as _struct_pb2
3
3
  from google.protobuf.internal import containers as _containers
4
+ from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
4
5
  from google.protobuf import descriptor as _descriptor
5
6
  from google.protobuf import message as _message
6
7
  from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
7
8
 
8
9
  DESCRIPTOR: _descriptor.FileDescriptor
9
10
 
10
- class GetJobSummaryRequest(_message.Message):
11
- __slots__ = ("job_id",)
12
- JOB_ID_FIELD_NUMBER: _ClassVar[int]
13
- job_id: str
14
- def __init__(self, job_id: _Optional[str] = ...) -> None: ...
15
-
16
- class GetJobSummaryResponse(_message.Message):
17
- __slots__ = ("job_summary_data",)
18
- JOB_SUMMARY_DATA_FIELD_NUMBER: _ClassVar[int]
19
- job_summary_data: _struct_pb2.Struct
20
- def __init__(self, job_summary_data: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ...
21
-
22
- class GetJobRequest(_message.Message):
23
- __slots__ = ("job_id",)
24
- JOB_ID_FIELD_NUMBER: _ClassVar[int]
25
- job_id: str
26
- def __init__(self, job_id: _Optional[str] = ...) -> None: ...
27
-
28
- class GetJobResponse(_message.Message):
29
- __slots__ = ("job_data",)
30
- JOB_DATA_FIELD_NUMBER: _ClassVar[int]
31
- job_data: _struct_pb2.Struct
32
- def __init__(self, job_data: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ...
33
-
34
- class ListJobsRequest(_message.Message):
35
- __slots__ = ("device_name", "job_name", "page", "limit", "sort_order")
11
+ class SelectionMode(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
12
+ __slots__ = ()
13
+ SELECTION_MODE_UNSPECIFIED: _ClassVar[SelectionMode]
14
+ SELECTION_MODE_ITEMS: _ClassVar[SelectionMode]
15
+ SELECTION_MODE_ALL: _ClassVar[SelectionMode]
16
+ SELECTION_MODE_UNSPECIFIED: SelectionMode
17
+ SELECTION_MODE_ITEMS: SelectionMode
18
+ SELECTION_MODE_ALL: SelectionMode
19
+
20
+ class SetSnapshotRequest(_message.Message):
21
+ __slots__ = ("device_name", "data")
22
+ DEVICE_NAME_FIELD_NUMBER: _ClassVar[int]
23
+ DATA_FIELD_NUMBER: _ClassVar[int]
24
+ device_name: str
25
+ data: _struct_pb2.Struct
26
+ def __init__(self, device_name: _Optional[str] = ..., data: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ...
27
+
28
+ class SetSnapshotResponse(_message.Message):
29
+ __slots__ = ("device_name", "version")
30
+ DEVICE_NAME_FIELD_NUMBER: _ClassVar[int]
31
+ VERSION_FIELD_NUMBER: _ClassVar[int]
32
+ device_name: str
33
+ version: int
34
+ def __init__(self, device_name: _Optional[str] = ..., version: _Optional[int] = ...) -> None: ...
35
+
36
+ class GetSnapshotRequest(_message.Message):
37
+ __slots__ = ("device_name", "version")
36
38
  DEVICE_NAME_FIELD_NUMBER: _ClassVar[int]
37
- JOB_NAME_FIELD_NUMBER: _ClassVar[int]
38
- PAGE_FIELD_NUMBER: _ClassVar[int]
39
- LIMIT_FIELD_NUMBER: _ClassVar[int]
40
- SORT_ORDER_FIELD_NUMBER: _ClassVar[int]
39
+ VERSION_FIELD_NUMBER: _ClassVar[int]
41
40
  device_name: str
42
- job_name: str
43
- page: int
44
- limit: int
45
- sort_order: int
46
- def __init__(self, device_name: _Optional[str] = ..., job_name: _Optional[str] = ..., page: _Optional[int] = ..., limit: _Optional[int] = ..., sort_order: _Optional[int] = ...) -> None: ...
47
-
48
- class ListJobsResponse(_message.Message):
49
- __slots__ = ("jobs", "total_pages")
50
- JOBS_FIELD_NUMBER: _ClassVar[int]
51
- TOTAL_PAGES_FIELD_NUMBER: _ClassVar[int]
52
- jobs: _containers.RepeatedCompositeFieldContainer[_struct_pb2.Struct]
53
- total_pages: int
54
- def __init__(self, jobs: _Optional[_Iterable[_Union[_struct_pb2.Struct, _Mapping]]] = ..., total_pages: _Optional[int] = ...) -> None: ...
41
+ version: int
42
+ def __init__(self, device_name: _Optional[str] = ..., version: _Optional[int] = ...) -> None: ...
43
+
44
+ class GetSnapshotResponse(_message.Message):
45
+ __slots__ = ("data",)
46
+ DATA_FIELD_NUMBER: _ClassVar[int]
47
+ data: _struct_pb2.Struct
48
+ def __init__(self, data: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ...
55
49
 
56
50
  class CreateRequest(_message.Message):
57
51
  __slots__ = ("app_name", "device_name", "device_data")
@@ -154,3 +148,33 @@ class GetAllDevicesMetadataResponse(_message.Message):
154
148
  METADATAS_FIELD_NUMBER: _ClassVar[int]
155
149
  metadatas: _containers.RepeatedCompositeFieldContainer[_struct_pb2.Struct]
156
150
  def __init__(self, metadatas: _Optional[_Iterable[_Union[_struct_pb2.Struct, _Mapping]]] = ...) -> None: ...
151
+
152
+ class MaskRequest(_message.Message):
153
+ __slots__ = ("mode", "device_name", "elements")
154
+ MODE_FIELD_NUMBER: _ClassVar[int]
155
+ DEVICE_NAME_FIELD_NUMBER: _ClassVar[int]
156
+ ELEMENTS_FIELD_NUMBER: _ClassVar[int]
157
+ mode: SelectionMode
158
+ device_name: str
159
+ elements: _containers.RepeatedScalarFieldContainer[str]
160
+ def __init__(self, mode: _Optional[_Union[SelectionMode, str]] = ..., device_name: _Optional[str] = ..., elements: _Optional[_Iterable[str]] = ...) -> None: ...
161
+
162
+ class MaskResponse(_message.Message):
163
+ __slots__ = ("done",)
164
+ DONE_FIELD_NUMBER: _ClassVar[int]
165
+ done: bool
166
+ def __init__(self, done: bool = ...) -> None: ...
167
+
168
+ class UnmaskRequest(_message.Message):
169
+ __slots__ = ("device_name", "elements")
170
+ DEVICE_NAME_FIELD_NUMBER: _ClassVar[int]
171
+ ELEMENTS_FIELD_NUMBER: _ClassVar[int]
172
+ device_name: str
173
+ elements: _containers.RepeatedScalarFieldContainer[str]
174
+ def __init__(self, device_name: _Optional[str] = ..., elements: _Optional[_Iterable[str]] = ...) -> None: ...
175
+
176
+ class UnmaskResponse(_message.Message):
177
+ __slots__ = ("done",)
178
+ DONE_FIELD_NUMBER: _ClassVar[int]
179
+ done: bool
180
+ def __init__(self, done: bool = ...) -> None: ...
@@ -35,26 +35,21 @@ class DeviceManagerServiceStub(object):
35
35
  request_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.UpdateRequest.SerializeToString,
36
36
  response_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.UpdateResponse.FromString,
37
37
  )
38
+ self.Mask = channel.unary_unary(
39
+ '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/Mask',
40
+ request_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.MaskRequest.SerializeToString,
41
+ response_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.MaskResponse.FromString,
42
+ )
43
+ self.Unmask = channel.unary_unary(
44
+ '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/Unmask',
45
+ request_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.UnmaskRequest.SerializeToString,
46
+ response_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.UnmaskResponse.FromString,
47
+ )
38
48
  self.Delete = channel.unary_unary(
39
49
  '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/Delete',
40
50
  request_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.DeleteRequest.SerializeToString,
41
51
  response_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.DeleteResponse.FromString,
42
52
  )
43
- self.ListJobs = channel.unary_unary(
44
- '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/ListJobs',
45
- request_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.ListJobsRequest.SerializeToString,
46
- response_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.ListJobsResponse.FromString,
47
- )
48
- self.GetJob = channel.unary_unary(
49
- '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/GetJob',
50
- request_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetJobRequest.SerializeToString,
51
- response_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetJobResponse.FromString,
52
- )
53
- self.GetJobSummary = channel.unary_unary(
54
- '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/GetJobSummary',
55
- request_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetJobSummaryRequest.SerializeToString,
56
- response_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetJobSummaryResponse.FromString,
57
- )
58
53
  self.GetMetadata = channel.unary_unary(
59
54
  '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/GetMetadata',
60
55
  request_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetMetadataRequest.SerializeToString,
@@ -65,6 +60,16 @@ class DeviceManagerServiceStub(object):
65
60
  request_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetAllDevicesMetadataRequest.SerializeToString,
66
61
  response_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetAllDevicesMetadataResponse.FromString,
67
62
  )
63
+ self.GetSnapshot = channel.unary_unary(
64
+ '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/GetSnapshot',
65
+ request_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetSnapshotRequest.SerializeToString,
66
+ response_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetSnapshotResponse.FromString,
67
+ )
68
+ self.SetSnapshot = channel.unary_unary(
69
+ '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/SetSnapshot',
70
+ request_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.SetSnapshotRequest.SerializeToString,
71
+ response_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.SetSnapshotResponse.FromString,
72
+ )
68
73
 
69
74
 
70
75
  class DeviceManagerServiceServicer(object):
@@ -99,43 +104,50 @@ class DeviceManagerServiceServicer(object):
99
104
  context.set_details('Method not implemented!')
100
105
  raise NotImplementedError('Method not implemented!')
101
106
 
102
- def Delete(self, request, context):
103
- """Delete the device and controller states.
107
+ def Mask(self, request, context):
108
+ """Mask elements on device.
104
109
  """
105
110
  context.set_code(grpc.StatusCode.UNIMPLEMENTED)
106
111
  context.set_details('Method not implemented!')
107
112
  raise NotImplementedError('Method not implemented!')
108
113
 
109
- def ListJobs(self, request, context):
110
- """Fetch all the experiments that have been run on the device.
114
+ def Unmask(self, request, context):
115
+ """Unmask elements on device.
111
116
  """
112
117
  context.set_code(grpc.StatusCode.UNIMPLEMENTED)
113
118
  context.set_details('Method not implemented!')
114
119
  raise NotImplementedError('Method not implemented!')
115
120
 
116
- def GetJob(self, request, context):
117
- """Fetch the results of a specific job executed on the device previously.
121
+ def Delete(self, request, context):
122
+ """Delete the device and controller states.
118
123
  """
119
124
  context.set_code(grpc.StatusCode.UNIMPLEMENTED)
120
125
  context.set_details('Method not implemented!')
121
126
  raise NotImplementedError('Method not implemented!')
122
127
 
123
- def GetJobSummary(self, request, context):
124
- """Fetch the results view of a previously executed job.
128
+ def GetMetadata(self, request, context):
129
+ """Fetch the metadata for the given device.
125
130
  """
126
131
  context.set_code(grpc.StatusCode.UNIMPLEMENTED)
127
132
  context.set_details('Method not implemented!')
128
133
  raise NotImplementedError('Method not implemented!')
129
134
 
130
- def GetMetadata(self, request, context):
131
- """Fetch the metadata for the given device.
135
+ def GetAllDevicesMetadata(self, request, context):
136
+ """Fetch information about all devices in this session.
132
137
  """
133
138
  context.set_code(grpc.StatusCode.UNIMPLEMENTED)
134
139
  context.set_details('Method not implemented!')
135
140
  raise NotImplementedError('Method not implemented!')
136
141
 
137
- def GetAllDevicesMetadata(self, request, context):
138
- """Fetch information about all devices in this session.
142
+ def GetSnapshot(self, request, context):
143
+ """Get a snapshot of a device
144
+ """
145
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
146
+ context.set_details('Method not implemented!')
147
+ raise NotImplementedError('Method not implemented!')
148
+
149
+ def SetSnapshot(self, request, context):
150
+ """Set a device state using a snapshot
139
151
  """
140
152
  context.set_code(grpc.StatusCode.UNIMPLEMENTED)
141
153
  context.set_details('Method not implemented!')
@@ -164,26 +176,21 @@ def add_DeviceManagerServiceServicer_to_server(servicer, server):
164
176
  request_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.UpdateRequest.FromString,
165
177
  response_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.UpdateResponse.SerializeToString,
166
178
  ),
179
+ 'Mask': grpc.unary_unary_rpc_method_handler(
180
+ servicer.Mask,
181
+ request_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.MaskRequest.FromString,
182
+ response_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.MaskResponse.SerializeToString,
183
+ ),
184
+ 'Unmask': grpc.unary_unary_rpc_method_handler(
185
+ servicer.Unmask,
186
+ request_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.UnmaskRequest.FromString,
187
+ response_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.UnmaskResponse.SerializeToString,
188
+ ),
167
189
  'Delete': grpc.unary_unary_rpc_method_handler(
168
190
  servicer.Delete,
169
191
  request_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.DeleteRequest.FromString,
170
192
  response_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.DeleteResponse.SerializeToString,
171
193
  ),
172
- 'ListJobs': grpc.unary_unary_rpc_method_handler(
173
- servicer.ListJobs,
174
- request_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.ListJobsRequest.FromString,
175
- response_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.ListJobsResponse.SerializeToString,
176
- ),
177
- 'GetJob': grpc.unary_unary_rpc_method_handler(
178
- servicer.GetJob,
179
- request_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetJobRequest.FromString,
180
- response_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetJobResponse.SerializeToString,
181
- ),
182
- 'GetJobSummary': grpc.unary_unary_rpc_method_handler(
183
- servicer.GetJobSummary,
184
- request_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetJobSummaryRequest.FromString,
185
- response_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetJobSummaryResponse.SerializeToString,
186
- ),
187
194
  'GetMetadata': grpc.unary_unary_rpc_method_handler(
188
195
  servicer.GetMetadata,
189
196
  request_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetMetadataRequest.FromString,
@@ -194,6 +201,16 @@ def add_DeviceManagerServiceServicer_to_server(servicer, server):
194
201
  request_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetAllDevicesMetadataRequest.FromString,
195
202
  response_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetAllDevicesMetadataResponse.SerializeToString,
196
203
  ),
204
+ 'GetSnapshot': grpc.unary_unary_rpc_method_handler(
205
+ servicer.GetSnapshot,
206
+ request_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetSnapshotRequest.FromString,
207
+ response_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetSnapshotResponse.SerializeToString,
208
+ ),
209
+ 'SetSnapshot': grpc.unary_unary_rpc_method_handler(
210
+ servicer.SetSnapshot,
211
+ request_deserializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.SetSnapshotRequest.FromString,
212
+ response_serializer=boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.SetSnapshotResponse.SerializeToString,
213
+ ),
197
214
  }
198
215
  generic_handler = grpc.method_handlers_generic_handler(
199
216
  'boulderopalscaleupsdk.protobuf.v1.DeviceManagerService', rpc_method_handlers)
@@ -274,7 +291,7 @@ class DeviceManagerService(object):
274
291
  insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
275
292
 
276
293
  @staticmethod
277
- def Delete(request,
294
+ def Mask(request,
278
295
  target,
279
296
  options=(),
280
297
  channel_credentials=None,
@@ -284,14 +301,14 @@ class DeviceManagerService(object):
284
301
  wait_for_ready=None,
285
302
  timeout=None,
286
303
  metadata=None):
287
- return grpc.experimental.unary_unary(request, target, '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/Delete',
288
- boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.DeleteRequest.SerializeToString,
289
- boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.DeleteResponse.FromString,
304
+ return grpc.experimental.unary_unary(request, target, '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/Mask',
305
+ boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.MaskRequest.SerializeToString,
306
+ boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.MaskResponse.FromString,
290
307
  options, channel_credentials,
291
308
  insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
292
309
 
293
310
  @staticmethod
294
- def ListJobs(request,
311
+ def Unmask(request,
295
312
  target,
296
313
  options=(),
297
314
  channel_credentials=None,
@@ -301,14 +318,14 @@ class DeviceManagerService(object):
301
318
  wait_for_ready=None,
302
319
  timeout=None,
303
320
  metadata=None):
304
- return grpc.experimental.unary_unary(request, target, '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/ListJobs',
305
- boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.ListJobsRequest.SerializeToString,
306
- boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.ListJobsResponse.FromString,
321
+ return grpc.experimental.unary_unary(request, target, '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/Unmask',
322
+ boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.UnmaskRequest.SerializeToString,
323
+ boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.UnmaskResponse.FromString,
307
324
  options, channel_credentials,
308
325
  insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
309
326
 
310
327
  @staticmethod
311
- def GetJob(request,
328
+ def Delete(request,
312
329
  target,
313
330
  options=(),
314
331
  channel_credentials=None,
@@ -318,14 +335,14 @@ class DeviceManagerService(object):
318
335
  wait_for_ready=None,
319
336
  timeout=None,
320
337
  metadata=None):
321
- return grpc.experimental.unary_unary(request, target, '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/GetJob',
322
- boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetJobRequest.SerializeToString,
323
- boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetJobResponse.FromString,
338
+ return grpc.experimental.unary_unary(request, target, '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/Delete',
339
+ boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.DeleteRequest.SerializeToString,
340
+ boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.DeleteResponse.FromString,
324
341
  options, channel_credentials,
325
342
  insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
326
343
 
327
344
  @staticmethod
328
- def GetJobSummary(request,
345
+ def GetMetadata(request,
329
346
  target,
330
347
  options=(),
331
348
  channel_credentials=None,
@@ -335,14 +352,14 @@ class DeviceManagerService(object):
335
352
  wait_for_ready=None,
336
353
  timeout=None,
337
354
  metadata=None):
338
- return grpc.experimental.unary_unary(request, target, '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/GetJobSummary',
339
- boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetJobSummaryRequest.SerializeToString,
340
- boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetJobSummaryResponse.FromString,
355
+ return grpc.experimental.unary_unary(request, target, '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/GetMetadata',
356
+ boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetMetadataRequest.SerializeToString,
357
+ boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetMetadataResponse.FromString,
341
358
  options, channel_credentials,
342
359
  insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
343
360
 
344
361
  @staticmethod
345
- def GetMetadata(request,
362
+ def GetAllDevicesMetadata(request,
346
363
  target,
347
364
  options=(),
348
365
  channel_credentials=None,
@@ -352,14 +369,14 @@ class DeviceManagerService(object):
352
369
  wait_for_ready=None,
353
370
  timeout=None,
354
371
  metadata=None):
355
- return grpc.experimental.unary_unary(request, target, '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/GetMetadata',
356
- boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetMetadataRequest.SerializeToString,
357
- boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetMetadataResponse.FromString,
372
+ return grpc.experimental.unary_unary(request, target, '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/GetAllDevicesMetadata',
373
+ boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetAllDevicesMetadataRequest.SerializeToString,
374
+ boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetAllDevicesMetadataResponse.FromString,
358
375
  options, channel_credentials,
359
376
  insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
360
377
 
361
378
  @staticmethod
362
- def GetAllDevicesMetadata(request,
379
+ def GetSnapshot(request,
363
380
  target,
364
381
  options=(),
365
382
  channel_credentials=None,
@@ -369,8 +386,25 @@ class DeviceManagerService(object):
369
386
  wait_for_ready=None,
370
387
  timeout=None,
371
388
  metadata=None):
372
- return grpc.experimental.unary_unary(request, target, '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/GetAllDevicesMetadata',
373
- boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetAllDevicesMetadataRequest.SerializeToString,
374
- boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetAllDevicesMetadataResponse.FromString,
389
+ return grpc.experimental.unary_unary(request, target, '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/GetSnapshot',
390
+ boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetSnapshotRequest.SerializeToString,
391
+ boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.GetSnapshotResponse.FromString,
392
+ options, channel_credentials,
393
+ insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
394
+
395
+ @staticmethod
396
+ def SetSnapshot(request,
397
+ target,
398
+ options=(),
399
+ channel_credentials=None,
400
+ call_credentials=None,
401
+ insecure=False,
402
+ compression=None,
403
+ wait_for_ready=None,
404
+ timeout=None,
405
+ metadata=None):
406
+ return grpc.experimental.unary_unary(request, target, '/boulderopalscaleupsdk.protobuf.v1.DeviceManagerService/SetSnapshot',
407
+ boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.SetSnapshotRequest.SerializeToString,
408
+ boulderopalscaleupsdk_dot_protobuf_dot_v1_dot_device__pb2.SetSnapshotResponse.FromString,
375
409
  options, channel_credentials,
376
410
  insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
@@ -0,0 +1,47 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # source: boulderopalscaleupsdk/protobuf/v1/job.proto
4
+ # Protobuf Python Version: 5.26.1
5
+ """Generated protocol buffer code."""
6
+ from google.protobuf import descriptor as _descriptor
7
+ from google.protobuf import descriptor_pool as _descriptor_pool
8
+ from google.protobuf import symbol_database as _symbol_database
9
+ from google.protobuf.internal import builder as _builder
10
+ # @@protoc_insertion_point(imports)
11
+
12
+ _sym_db = _symbol_database.Default()
13
+
14
+
15
+ from google.api import annotations_pb2 as google_dot_api_dot_annotations__pb2
16
+ from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
17
+
18
+
19
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n+boulderopalscaleupsdk/protobuf/v1/job.proto\x12!boulderopalscaleupsdk.protobuf.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x1cgoogle/protobuf/struct.proto\"*\n\x11GetSummaryRequest\x12\x15\n\x06job_id\x18\x01 \x01(\tR\x05jobId\"W\n\x12GetSummaryResponse\x12\x41\n\x10job_summary_data\x18\x01 \x01(\x0b\x32\x17.google.protobuf.StructR\x0ejobSummaryData\"#\n\nGetRequest\x12\x15\n\x06job_id\x18\x01 \x01(\tR\x05jobId\"A\n\x0bGetResponse\x12\x32\n\x08job_data\x18\x01 \x01(\x0b\x32\x17.google.protobuf.StructR\x07jobData\"\x92\x01\n\x0bListRequest\x12\x1f\n\x0b\x64\x65vice_name\x18\x01 \x01(\tR\ndeviceName\x12\x19\n\x08job_name\x18\x02 \x01(\tR\x07jobName\x12\x12\n\x04page\x18\x03 \x01(\x05R\x04page\x12\x14\n\x05limit\x18\x04 \x01(\x05R\x05limit\x12\x1d\n\nsort_order\x18\x05 \x01(\x05R\tsortOrder\"\\\n\x0cListResponse\x12+\n\x04jobs\x18\x01 \x03(\x0b\x32\x17.google.protobuf.StructR\x04jobs\x12\x1f\n\x0btotal_pages\x18\x02 \x01(\x05R\ntotalPages2\xb9\x03\n\x11JobManagerService\x12\x80\x01\n\x04List\x12..boulderopalscaleupsdk.protobuf.v1.ListRequest\x1a/.boulderopalscaleupsdk.protobuf.v1.ListResponse\"\x17\x82\xd3\xe4\x93\x02\x11\"\x0c/v1/job/list:\x01*\x12\x80\x01\n\x03Get\x12-.boulderopalscaleupsdk.protobuf.v1.GetRequest\x1a..boulderopalscaleupsdk.protobuf.v1.GetResponse\"\x1a\x82\xd3\xe4\x93\x02\x14\"\x0f/v1/job/get_job:\x01*\x12\x9d\x01\n\nGetSummary\x12\x34.boulderopalscaleupsdk.protobuf.v1.GetSummaryRequest\x1a\x35.boulderopalscaleupsdk.protobuf.v1.GetSummaryResponse\"\"\x82\xd3\xe4\x93\x02\x1c\"\x17/v1/job/get_job_summary:\x01*B\xa7\x02\n%com.boulderopalscaleupsdk.protobuf.v1B\x08JobProtoP\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')
20
+
21
+ _globals = globals()
22
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
23
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'boulderopalscaleupsdk.protobuf.v1.job_pb2', _globals)
24
+ if not _descriptor._USE_C_DESCRIPTORS:
25
+ _globals['DESCRIPTOR']._loaded_options = None
26
+ _globals['DESCRIPTOR']._serialized_options = b'\n%com.boulderopalscaleupsdk.protobuf.v1B\010JobProtoP\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'
27
+ _globals['_JOBMANAGERSERVICE'].methods_by_name['List']._loaded_options = None
28
+ _globals['_JOBMANAGERSERVICE'].methods_by_name['List']._serialized_options = b'\202\323\344\223\002\021\"\014/v1/job/list:\001*'
29
+ _globals['_JOBMANAGERSERVICE'].methods_by_name['Get']._loaded_options = None
30
+ _globals['_JOBMANAGERSERVICE'].methods_by_name['Get']._serialized_options = b'\202\323\344\223\002\024\"\017/v1/job/get_job:\001*'
31
+ _globals['_JOBMANAGERSERVICE'].methods_by_name['GetSummary']._loaded_options = None
32
+ _globals['_JOBMANAGERSERVICE'].methods_by_name['GetSummary']._serialized_options = b'\202\323\344\223\002\034\"\027/v1/job/get_job_summary:\001*'
33
+ _globals['_GETSUMMARYREQUEST']._serialized_start=142
34
+ _globals['_GETSUMMARYREQUEST']._serialized_end=184
35
+ _globals['_GETSUMMARYRESPONSE']._serialized_start=186
36
+ _globals['_GETSUMMARYRESPONSE']._serialized_end=273
37
+ _globals['_GETREQUEST']._serialized_start=275
38
+ _globals['_GETREQUEST']._serialized_end=310
39
+ _globals['_GETRESPONSE']._serialized_start=312
40
+ _globals['_GETRESPONSE']._serialized_end=377
41
+ _globals['_LISTREQUEST']._serialized_start=380
42
+ _globals['_LISTREQUEST']._serialized_end=526
43
+ _globals['_LISTRESPONSE']._serialized_start=528
44
+ _globals['_LISTRESPONSE']._serialized_end=620
45
+ _globals['_JOBMANAGERSERVICE']._serialized_start=623
46
+ _globals['_JOBMANAGERSERVICE']._serialized_end=1064
47
+ # @@protoc_insertion_point(module_scope)
@@ -0,0 +1,54 @@
1
+ from google.api import annotations_pb2 as _annotations_pb2
2
+ from google.protobuf import struct_pb2 as _struct_pb2
3
+ from google.protobuf.internal import containers as _containers
4
+ from google.protobuf import descriptor as _descriptor
5
+ from google.protobuf import message as _message
6
+ from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
7
+
8
+ DESCRIPTOR: _descriptor.FileDescriptor
9
+
10
+ class GetSummaryRequest(_message.Message):
11
+ __slots__ = ("job_id",)
12
+ JOB_ID_FIELD_NUMBER: _ClassVar[int]
13
+ job_id: str
14
+ def __init__(self, job_id: _Optional[str] = ...) -> None: ...
15
+
16
+ class GetSummaryResponse(_message.Message):
17
+ __slots__ = ("job_summary_data",)
18
+ JOB_SUMMARY_DATA_FIELD_NUMBER: _ClassVar[int]
19
+ job_summary_data: _struct_pb2.Struct
20
+ def __init__(self, job_summary_data: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ...
21
+
22
+ class GetRequest(_message.Message):
23
+ __slots__ = ("job_id",)
24
+ JOB_ID_FIELD_NUMBER: _ClassVar[int]
25
+ job_id: str
26
+ def __init__(self, job_id: _Optional[str] = ...) -> None: ...
27
+
28
+ class GetResponse(_message.Message):
29
+ __slots__ = ("job_data",)
30
+ JOB_DATA_FIELD_NUMBER: _ClassVar[int]
31
+ job_data: _struct_pb2.Struct
32
+ def __init__(self, job_data: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ...
33
+
34
+ class ListRequest(_message.Message):
35
+ __slots__ = ("device_name", "job_name", "page", "limit", "sort_order")
36
+ DEVICE_NAME_FIELD_NUMBER: _ClassVar[int]
37
+ JOB_NAME_FIELD_NUMBER: _ClassVar[int]
38
+ PAGE_FIELD_NUMBER: _ClassVar[int]
39
+ LIMIT_FIELD_NUMBER: _ClassVar[int]
40
+ SORT_ORDER_FIELD_NUMBER: _ClassVar[int]
41
+ device_name: str
42
+ job_name: str
43
+ page: int
44
+ limit: int
45
+ sort_order: int
46
+ def __init__(self, device_name: _Optional[str] = ..., job_name: _Optional[str] = ..., page: _Optional[int] = ..., limit: _Optional[int] = ..., sort_order: _Optional[int] = ...) -> None: ...
47
+
48
+ class ListResponse(_message.Message):
49
+ __slots__ = ("jobs", "total_pages")
50
+ JOBS_FIELD_NUMBER: _ClassVar[int]
51
+ TOTAL_PAGES_FIELD_NUMBER: _ClassVar[int]
52
+ jobs: _containers.RepeatedCompositeFieldContainer[_struct_pb2.Struct]
53
+ total_pages: int
54
+ def __init__(self, jobs: _Optional[_Iterable[_Union[_struct_pb2.Struct, _Mapping]]] = ..., total_pages: _Optional[int] = ...) -> None: ...