QuLab 2.7.6__cp311-cp311-macosx_10_9_universal2.whl → 2.7.7__cp311-cp311-macosx_10_9_universal2.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.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.2
2
2
  Name: QuLab
3
- Version: 2.7.6
3
+ Version: 2.7.7
4
4
  Summary: contral instruments and manage data
5
5
  Author-email: feihoo87 <feihoo87@gmail.com>
6
6
  Maintainer-email: feihoo87 <feihoo87@gmail.com>
@@ -1,20 +1,20 @@
1
- qulab/__init__.py,sha256=tKRZe2WafDGcwbu7ddlMaePerN842aRbxcrECsu5pas,286
1
+ qulab/__init__.py,sha256=KJcUcZ5qXY6wlAoirzK_B-dgtDjsLmOE671v3gcXO_c,286
2
2
  qulab/__main__.py,sha256=fjaRSL_uUjNIzBGNgjlGswb9TJ2VD5qnkZHW3hItrD4,68
3
3
  qulab/dicttree.py,sha256=tRRMpGZYVOLw0TEByE3_2Ss8FdOmzuGL9e1DWbs8qoY,13684
4
- qulab/fun.cpython-311-darwin.so,sha256=Nf_7Ra789hwgSaL6hlas-Tlkd1UxO0ZkHFbuM1GopZI,126848
5
- qulab/typing.py,sha256=5xCLfrp1aZpKpDy4p2arbFszw2eK3hGUjZa-XSvC_-8,69
4
+ qulab/fun.cpython-311-darwin.so,sha256=gdwo5kR7RsUYV69Dp40GEMDtlRvD7QhmrQRED08p3Ow,126848
5
+ qulab/typing.py,sha256=vg62sGqxuD9CI5677ejlzAmf2fVdAESZCQjAE_xSxPg,69
6
6
  qulab/utils.py,sha256=JIXMSmZU0uYfKG_tzawpK7vRNPRir_hJE8JlqkVLX2o,1260
7
- qulab/version.py,sha256=ElxLmcD9H2GsJrWa-NsuGSxQyl8yztVfJ8oaA66f8oQ,21
7
+ qulab/version.py,sha256=TXxZaBvn1Y_BQ2YYBZ0iedwKmA3iOfYiZehFXuAoky0,21
8
8
  qulab/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
9
9
  qulab/cli/commands.py,sha256=tgDIkkeIoasQXAifJZ6NU8jDgpNgb2a-B0C4nF0evrE,559
10
10
  qulab/cli/config.py,sha256=SdNmWzweWAdyk8M2oKYhMxnkaJ0qIayPlsLGCNlVqME,3108
11
11
  qulab/executor/__init__.py,sha256=LosPzOMaljSZY1thy_Fxtbrgq7uubJszMABEB7oM7tU,101
12
- qulab/executor/cli.py,sha256=n0wl27xaiMWg18rVDOaKgLTEXaN0tSv4whTUQxvzBoM,8932
13
- qulab/executor/load.py,sha256=RJpgDuxNc3sUzr65K6a4vO-Qa72_lYsJGrPZDfw9Rus,18077
14
- qulab/executor/schedule.py,sha256=jBUK0vXeu-gbieSaH245exlGu3TW6vg9o0GIEF43IvA,16796
15
- qulab/executor/storage.py,sha256=k8rcyUHaAixMtWRTNfQrjFhingYqBb0fJZwHBLVE_UM,11002
16
- qulab/executor/transform.py,sha256=kaDJePBhVc32gZkQN8YBuA69lYYW9YhDZmCnUKaWuQw,2189
17
- qulab/executor/utils.py,sha256=XDk3duNrqDsCqZyZUmPqIAIIiykiafcOMBpy9NVAOAw,5686
12
+ qulab/executor/cli.py,sha256=gGka2M6xccfM8facsIJ2qZ1y2Yx8C4BRhc1JG6nK9mo,8932
13
+ qulab/executor/load.py,sha256=4FY_SwumLDUewC265v4j_ZGGpfYOgH4c8PtglYcWpBw,18077
14
+ qulab/executor/schedule.py,sha256=DHQ5dI5YX8_frWplOoLEb9htcfM5-mikiSBNSPWT1io,16725
15
+ qulab/executor/storage.py,sha256=sBD-aNvj29l5HtoTpk_627qarZkPn33F-hcc80AuF6k,11079
16
+ qulab/executor/transform.py,sha256=s0kxWQx8Sr9pMIQke1BLNM6KqrSogAkjB6Zkapl8YSU,2189
17
+ qulab/executor/utils.py,sha256=cF6-2jlvlHyTjNHdxXKG04Fjfm3_3wfDQAF1G8DQphk,5686
18
18
  qulab/monitor/__init__.py,sha256=nTHelnDpxRS_fl_B38TsN0njgq8eVTEz9IAnN3NbDlM,42
19
19
  qulab/monitor/__main__.py,sha256=w3yUcqq195LzSnXTkQcuC1RSFRhy4oQ_PEBmucXguME,97
20
20
  qulab/monitor/config.py,sha256=fQ5JcsMApKc1UwANEnIvbDQZl8uYW0tle92SaYtX9lI,744
@@ -94,9 +94,9 @@ qulab/visualization/plot_seq.py,sha256=UWTS6p9nfX_7B8ehcYo6UnSTUCjkBsNU9jiOeW2ca
94
94
  qulab/visualization/qdat.py,sha256=ZeevBYWkzbww4xZnsjHhw7wRorJCBzbG0iEu-XQB4EA,5735
95
95
  qulab/visualization/rot3d.py,sha256=lMrEJlRLwYe6NMBlGkKYpp_V9CTipOAuDy6QW_cQK00,734
96
96
  qulab/visualization/widgets.py,sha256=6KkiTyQ8J-ei70LbPQZAK35wjktY47w2IveOa682ftA,3180
97
- QuLab-2.7.6.dist-info/LICENSE,sha256=PRzIKxZtpQcH7whTG6Egvzl1A0BvnSf30tmR2X2KrpA,1065
98
- QuLab-2.7.6.dist-info/METADATA,sha256=CUCFSSV87lDFOp15xOhzDpvmL3P7TMje6GOlrckku_E,3698
99
- QuLab-2.7.6.dist-info/WHEEL,sha256=K10eKSN6_vzvMOgXxWbVOQNR7Orfl6gBTCpCI8bcYx4,114
100
- QuLab-2.7.6.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
101
- QuLab-2.7.6.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
102
- QuLab-2.7.6.dist-info/RECORD,,
97
+ QuLab-2.7.7.dist-info/LICENSE,sha256=PRzIKxZtpQcH7whTG6Egvzl1A0BvnSf30tmR2X2KrpA,1065
98
+ QuLab-2.7.7.dist-info/METADATA,sha256=uhQQtVYRkuDdN8Inh-gnDFZDTekIqQRrQEo5V9hPD1A,3698
99
+ QuLab-2.7.7.dist-info/WHEEL,sha256=K10eKSN6_vzvMOgXxWbVOQNR7Orfl6gBTCpCI8bcYx4,114
100
+ QuLab-2.7.7.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
101
+ QuLab-2.7.7.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
102
+ QuLab-2.7.7.dist-info/RECORD,,
qulab/__init__.py CHANGED
@@ -1,5 +1,5 @@
1
- from .executor.storage import find_result
2
- from .executor.storage import get_result_by_index as get_result
1
+ from .executor.storage import find_report
2
+ from .executor.storage import get_report_by_index as get_report
3
3
  from .executor.utils import debug_analyze
4
4
  from .scan import Scan, get_record, load_record, lookup, lookup_list
5
5
  from .version import __version__
qulab/executor/cli.py CHANGED
@@ -149,7 +149,7 @@ def get(key, api):
149
149
 
150
150
  @click.command()
151
151
  @click.argument('workflow')
152
- @click.option('--plot', '-p', is_flag=True, help='Plot the result.')
152
+ @click.option('--plot', '-p', is_flag=True, help='Plot the report.')
153
153
  @click.option('--no-dependents',
154
154
  '-n',
155
155
  is_flag=True,
@@ -221,7 +221,7 @@ def run(workflow, code, data, api, plot, no_dependents, retry, freeze):
221
221
  @click.command()
222
222
  @click.argument('workflow')
223
223
  @click.option('--retry', '-r', default=1, type=int, help='Retry times.')
224
- @click.option('--plot', '-p', is_flag=True, help='Plot the result.')
224
+ @click.option('--plot', '-p', is_flag=True, help='Plot the report.')
225
225
  @log_options
226
226
  @command_option('maintain')
227
227
  def maintain(workflow, code, data, api, retry, plot):
qulab/executor/load.py CHANGED
@@ -15,7 +15,7 @@ from typing import Any
15
15
 
16
16
  from loguru import logger
17
17
 
18
- from .storage import Result
18
+ from .storage import Report
19
19
 
20
20
 
21
21
  class SetConfigWorkflow():
@@ -29,7 +29,7 @@ class SetConfigWorkflow():
29
29
  def depends(self):
30
30
  return []
31
31
 
32
- def check_state(self, history: Result) -> bool:
32
+ def check_state(self, history: Report) -> bool:
33
33
  from . import transform
34
34
  try:
35
35
  return self._equal(history.parameters[self.key],
@@ -45,21 +45,21 @@ class SetConfigWorkflow():
45
45
  value = eval(input(f'"{self.key}": '))
46
46
  return value
47
47
 
48
- def analyze(self, result: Result, history: Result):
49
- result.state = 'OK'
50
- result.parameters = {self.key: result.data}
51
- return result
48
+ def analyze(self, report: Report, history: Report):
49
+ report.state = 'OK'
50
+ report.parameters = {self.key: report.data}
51
+ return report
52
52
 
53
53
  def check(self):
54
54
  return self.calibrate()
55
55
 
56
- def check_analyze(self, result: Result, history: Result | None):
56
+ def check_analyze(self, report: Report, history: Report | None):
57
57
  if self.check_state(history):
58
- result.state = 'OK'
59
- result.parameters = {self.key: history.data}
58
+ report.state = 'OK'
59
+ report.parameters = {self.key: history.data}
60
60
  else:
61
- result.state = 'Outdated'
62
- return result
61
+ report.state = 'Outdated'
62
+ return report
63
63
 
64
64
  @staticmethod
65
65
  def _equal(a, b):
@@ -7,28 +7,28 @@ from pathlib import Path
7
7
  from loguru import logger
8
8
 
9
9
  from .load import WorkflowType, get_dependents
10
- from .storage import (Result, find_result, get_heads, renew_result,
11
- revoke_result, save_result)
10
+ from .storage import (Report, find_report, get_heads, renew_report,
11
+ revoke_report, save_report)
12
12
  from .transform import current_config, obey_the_oracle, update_parameters
13
13
 
14
14
  __session_id = None
15
15
  __session_cache = {}
16
16
 
17
17
 
18
- def set_cache(session_id, key, result: Result):
18
+ def set_cache(session_id, key, report: Report):
19
19
  global __session_id
20
20
  if __session_id is None:
21
21
  __session_id = session_id
22
22
  if __session_id != session_id:
23
23
  __session_cache.clear()
24
- if result.workflow.startswith('cfg:'):
25
- __session_cache[key] = result
24
+ if report.workflow.startswith('cfg:'):
25
+ __session_cache[key] = report
26
26
  else:
27
- __session_cache[key] = result.base_path, result.path
27
+ __session_cache[key] = report.base_path, report.path
28
28
 
29
29
 
30
- def get_cache(session_id, key) -> Result:
31
- from .storage import load_result
30
+ def get_cache(session_id, key) -> Report:
31
+ from .storage import load_report
32
32
  global __session_id
33
33
  if __session_id is None or __session_id != session_id:
34
34
  return None
@@ -37,8 +37,8 @@ def get_cache(session_id, key) -> Result:
37
37
  return None
38
38
  if isinstance(index, tuple):
39
39
  base_path, path = index
40
- return load_result(base_path, path)
41
- elif isinstance(index, Result):
40
+ return load_report(base_path, path)
41
+ elif isinstance(index, Report):
42
42
  return index
43
43
  else:
44
44
  return None
@@ -56,14 +56,14 @@ def is_pickleable(obj) -> bool:
56
56
  return False
57
57
 
58
58
 
59
- def veryfy_analyzed_result(result: Result, script: str, method: str):
60
- if not isinstance(result, Result):
61
- raise TypeError(f'"{script}" : "{method}" must return a Result object')
62
- if not is_pickleable(result.parameters):
59
+ def veryfy_analyzed_report(report: Report, script: str, method: str):
60
+ if not isinstance(report, Report):
61
+ raise TypeError(f'"{script}" : "{method}" must return a Report object')
62
+ if not is_pickleable(report.parameters):
63
63
  raise TypeError(
64
64
  f'"{script}" : "{method}" return not pickleable data in .parameters'
65
65
  )
66
- if not is_pickleable(result.other_infomation):
66
+ if not is_pickleable(report.other_infomation):
67
67
  raise TypeError(
68
68
  f'"{script}" : "{method}" return not pickleable data in .other_infomation'
69
69
  )
@@ -80,8 +80,8 @@ def check_state(workflow: WorkflowType, code_path: str | Path,
80
80
  4. All dependencies pass check state.
81
81
  """
82
82
  logger.debug(f'check_state: "{workflow.__workflow_id__}"')
83
- result = find_result(workflow.__workflow_id__, state_path)
84
- if not result:
83
+ report = find_report(workflow.__workflow_id__, state_path)
84
+ if not report:
85
85
  logger.debug(
86
86
  f'check_state failed: No history found for "{workflow.__workflow_id__}"'
87
87
  )
@@ -90,23 +90,23 @@ def check_state(workflow: WorkflowType, code_path: str | Path,
90
90
  logger.debug(
91
91
  f'check_state: "{workflow.__workflow_id__}" has custom check_state method'
92
92
  )
93
- return workflow.check_state(result)
93
+ return workflow.check_state(report)
94
94
  if workflow.__timeout__ is not None and datetime.now(
95
- ) > result.checked_time + timedelta(seconds=workflow.__timeout__):
95
+ ) > report.checked_time + timedelta(seconds=workflow.__timeout__):
96
96
  logger.debug(
97
97
  f'check_state failed: "{workflow.__workflow_id__}" has expired')
98
98
  return False
99
- if not result.in_spec:
99
+ if not report.in_spec:
100
100
  logger.debug(
101
101
  f'check_state failed: "{workflow.__workflow_id__}" is out of spec')
102
102
  return False
103
- if result.bad_data:
103
+ if report.bad_data:
104
104
  logger.debug(
105
105
  f'check_state failed: "{workflow.__workflow_id__}" has bad data')
106
106
  return False
107
107
  for n in get_dependents(workflow, code_path):
108
- r = find_result(n.__workflow_id__, state_path)
109
- if r is None or r.checked_time > result.checked_time:
108
+ r = find_report(n.__workflow_id__, state_path)
109
+ if r is None or r.checked_time > report.checked_time:
110
110
  logger.debug(
111
111
  f'check_state failed: "{workflow.__workflow_id__}" has outdated dependencies'
112
112
  )
@@ -120,71 +120,115 @@ def check_state(workflow: WorkflowType, code_path: str | Path,
120
120
  return True
121
121
 
122
122
 
123
- def call_analyzer(node,
124
- result: Result,
125
- history: Result | None,
123
+ def call_analyzer(node: WorkflowType,
124
+ report: Report,
125
+ history: Report | None,
126
126
  check=False,
127
- plot=False) -> Result:
127
+ plot=False) -> Report:
128
128
  if check:
129
- result = node.check_analyze(result, history=history)
130
- veryfy_analyzed_result(result, node.__workflow_id__, "check_analyze")
131
- result.fully_calibrated = False
129
+ report = node.check_analyze(report, history=history)
130
+ veryfy_analyzed_report(report, node.__workflow_id__, "check_analyze")
131
+ report.fully_calibrated = False
132
132
  else:
133
- result = node.analyze(result, history=history)
134
- veryfy_analyzed_result(result, node.__workflow_id__, "analyze")
133
+ report = node.analyze(report, history=history)
134
+ veryfy_analyzed_report(report, node.__workflow_id__, "analyze")
135
135
  if hasattr(node, 'oracle') and callable(node.oracle):
136
136
  logger.debug(
137
137
  f'"{node.__workflow_id__}" has oracle method, calling ...')
138
138
  try:
139
- result = node.oracle(result,
139
+ report = node.oracle(report,
140
140
  history=history,
141
- system_state=get_heads(result.base_path))
141
+ system_state=get_heads(report.base_path))
142
142
  except Exception as e:
143
143
  logger.exception(e)
144
- result.oracle = {}
145
- if not isinstance(result, Result):
144
+ report.oracle = {}
145
+ if not isinstance(report, Report):
146
146
  raise TypeError(
147
- f'"{node.__workflow_id__}" : function "oracle" must return a Result object'
147
+ f'"{node.__workflow_id__}" : function "oracle" must return a Report object'
148
148
  )
149
- if not is_pickleable(result.oracle):
149
+ if not is_pickleable(report.oracle):
150
150
  raise TypeError(
151
151
  f'"{node.__workflow_id__}" : function "oracle" return not pickleable data'
152
152
  )
153
- result.fully_calibrated = True
153
+ report.fully_calibrated = True
154
154
  if plot:
155
- call_plot(node, result)
156
- return result
155
+ call_plot(node, report)
156
+ return report
157
157
 
158
158
 
159
159
  @logger.catch()
160
- def call_plot(node, result: Result, check=False):
160
+ def call_plot(node: WorkflowType, report: Report, check=False):
161
161
  if hasattr(node, 'plot') and callable(node.plot):
162
- node.plot(result)
162
+ node.plot(report)
163
+
164
+
165
+ def call_check(workflow: WorkflowType, session_id: str, state_path: Path):
166
+ report = get_cache(session_id, (workflow, 'check'))
167
+ if report is not None:
168
+ logger.debug(f'Cache hit for "{workflow.__workflow_id__}:check"')
169
+ return report
170
+
171
+ data = workflow.check()
172
+ if not is_pickleable(data):
173
+ raise TypeError(
174
+ f'"{workflow.__workflow_id__}" : "check" return not pickleable data'
175
+ )
176
+ report = Report(workflow=workflow.__workflow_id__,
177
+ data=data,
178
+ config_path=current_config(state_path),
179
+ base_path=state_path,
180
+ heads=get_heads(state_path))
181
+
182
+ save_report(workflow.__workflow_id__,
183
+ report,
184
+ state_path,
185
+ refresh_heads=False)
186
+
187
+ set_cache(session_id, (workflow, 'check'), report)
188
+ return report
189
+
190
+
191
+ def call_calibrate(workflow: WorkflowType, session_id: str, state_path: Path):
192
+ report = get_cache(session_id, (workflow, 'calibrate'))
193
+ if report is not None:
194
+ logger.debug(f'Cache hit for "{workflow.__workflow_id__}:calibrate"')
195
+ return report
196
+
197
+ data = workflow.calibrate()
198
+ if not is_pickleable(data):
199
+ raise TypeError(
200
+ f'"{workflow.__workflow_id__}" : "calibrate" return not pickleable data'
201
+ )
202
+ report = Report(workflow=workflow.__workflow_id__,
203
+ data=data,
204
+ config_path=current_config(state_path),
205
+ base_path=state_path,
206
+ heads=get_heads(state_path))
207
+
208
+ save_report(workflow.__workflow_id__, report, state_path)
209
+
210
+ set_cache(session_id, (workflow, 'calibrate'), report)
211
+ return report
163
212
 
164
213
 
165
214
  def check_data(workflow: WorkflowType, code_path: str | Path,
166
- state_path: str | Path, plot: bool, session_id: str) -> Result:
215
+ state_path: str | Path, plot: bool, session_id: str) -> Report:
167
216
  """
168
217
  check data answers two questions:
169
218
  Is the parameter associated with this cal in spec,
170
219
  and is the cal scan working as expected?
171
220
  """
172
- result = get_cache(session_id, (workflow, 'check_data'))
173
- if result is not None:
174
- logger.debug(f'Cache hit for "{workflow.__workflow_id__}:check_data"')
175
- return result
176
-
177
- history = find_result(workflow.__workflow_id__, state_path)
221
+ history = find_report(workflow.__workflow_id__, state_path)
178
222
 
179
223
  if history is None:
180
224
  logger.debug(f'No history found for "{workflow.__workflow_id__}"')
181
- result = Result(workflow=workflow.__workflow_id__,
225
+ report = Report(workflow=workflow.__workflow_id__,
182
226
  config_path=current_config(state_path),
183
227
  base_path=state_path,
184
228
  heads=get_heads(state_path))
185
- result.in_spec = False
186
- result.bad_data = False
187
- return result
229
+ report.in_spec = False
230
+ report.bad_data = False
231
+ return report
188
232
 
189
233
  if history.bad_data:
190
234
  logger.debug(
@@ -202,92 +246,59 @@ def check_data(workflow: WorkflowType, code_path: str | Path,
202
246
  workflow, 'check_analyze') and callable(workflow.check_analyze):
203
247
  logger.debug(
204
248
  f'Checking "{workflow.__workflow_id__}" with "check" method ...')
205
- data = workflow.check()
206
- if not is_pickleable(data):
207
- raise TypeError(
208
- f'"{workflow.__workflow_id__}" : "check" return not pickleable data'
209
- )
210
- result = Result(workflow=workflow.__workflow_id__,
211
- data=data,
212
- config_path=current_config(state_path),
213
- base_path=state_path,
214
- heads=get_heads(state_path))
215
- #save_result(workflow.__workflow_id__, result, state_path)
249
+
250
+ report = call_check(workflow, session_id, state_path)
216
251
 
217
252
  logger.debug(f'Checked "{workflow.__workflow_id__}" !')
218
- result = call_analyzer(workflow,
219
- result,
253
+ report = call_analyzer(workflow,
254
+ report,
220
255
  history,
221
256
  check=True,
222
257
  plot=plot)
223
- if result.in_spec:
258
+ if report.in_spec:
224
259
  logger.debug(
225
- f'"{workflow.__workflow_id__}": checked in spec, renewing result'
260
+ f'"{workflow.__workflow_id__}": checked in spec, renewing report'
226
261
  )
227
- renew_result(workflow.__workflow_id__, state_path)
262
+ renew_report(workflow.__workflow_id__, report, state_path)
228
263
  else:
229
264
  logger.debug(
230
- f'"{workflow.__workflow_id__}": checked out of spec, revoking result'
265
+ f'"{workflow.__workflow_id__}": checked out of spec, revoking report'
231
266
  )
232
- revoke_result(workflow.__workflow_id__, state_path)
267
+ revoke_report(workflow.__workflow_id__, report, state_path)
233
268
  else:
234
269
  logger.debug(
235
270
  f'Checking "{workflow.__workflow_id__}" with "calibrate" method ...'
236
271
  )
237
- data = workflow.calibrate()
238
- if not is_pickleable(data):
239
- raise TypeError(
240
- f'"{workflow.__workflow_id__}" : "calibrate" return not pickleable data'
241
- )
242
- result = Result(workflow=workflow.__workflow_id__,
243
- data=data,
244
- config_path=current_config(state_path),
245
- base_path=state_path,
246
- heads=get_heads(state_path))
247
- save_result(workflow.__workflow_id__, result, state_path)
272
+
273
+ report = call_calibrate(workflow, session_id, state_path)
248
274
 
249
275
  logger.debug(f'Calibrated "{workflow.__workflow_id__}" !')
250
- result = call_analyzer(workflow,
251
- result,
276
+ report = call_analyzer(workflow,
277
+ report,
252
278
  history,
253
279
  check=False,
254
280
  plot=plot)
255
- save_result(workflow.__workflow_id__,
256
- result,
281
+ save_report(workflow.__workflow_id__,
282
+ report,
257
283
  state_path,
258
284
  overwrite=True)
259
-
260
- set_cache(session_id, (workflow, 'check_data'), result)
261
- return result
285
+ return report
262
286
 
263
287
 
264
288
  def calibrate(workflow: WorkflowType, code_path: str | Path,
265
- state_path: str | Path, plot: bool, session_id: str) -> Result:
266
- result = get_cache(session_id, (workflow, 'calibrate'))
267
- if result is not None:
268
- logger.debug(f'Cache hit for "{workflow.__workflow_id__}:calibrate"')
269
- return result
270
-
271
- history = find_result(workflow.__workflow_id__, state_path)
289
+ state_path: str | Path, plot: bool, session_id: str) -> Report:
290
+ history = find_report(workflow.__workflow_id__, state_path)
272
291
 
273
292
  logger.debug(f'Calibrating "{workflow.__workflow_id__}" ...')
274
- data = workflow.calibrate()
275
- if not is_pickleable(data):
276
- raise TypeError(
277
- f'"{workflow.__workflow_id__}" : "calibrate" return not pickleable data'
278
- )
279
- result = Result(workflow=workflow.__workflow_id__,
280
- data=data,
281
- config_path=current_config(state_path),
282
- base_path=state_path,
283
- heads=get_heads(state_path))
284
- save_result(workflow.__workflow_id__, result, state_path)
293
+
294
+ report = call_calibrate(workflow, session_id, state_path)
295
+
285
296
  logger.debug(f'Calibrated "{workflow.__workflow_id__}" !')
286
- result = call_analyzer(workflow, result, history, check=False, plot=plot)
287
- save_result(workflow.__workflow_id__, result, state_path, overwrite=True)
288
297
 
289
- set_cache(session_id, (workflow, 'calibrate'), result)
290
- return result
298
+ report = call_analyzer(workflow, report, history, check=False, plot=plot)
299
+
300
+ save_report(workflow.__workflow_id__, report, state_path, overwrite=True)
301
+ return report
291
302
 
292
303
 
293
304
  def diagnose(workflow: WorkflowType, code_path: str | Path,
@@ -297,16 +308,16 @@ def diagnose(workflow: WorkflowType, code_path: str | Path,
297
308
  '''
298
309
  logger.debug(f'diagnose "{workflow.__workflow_id__}"')
299
310
  # check_data
300
- result = check_data(workflow, code_path, state_path, plot, session_id)
311
+ report = check_data(workflow, code_path, state_path, plot, session_id)
301
312
  # in spec case
302
- if result.in_spec:
313
+ if report.in_spec:
303
314
  logger.debug(
304
315
  f'"{workflow.__workflow_id__}": Checked! In spec, no need to diagnose'
305
316
  )
306
317
  return False
307
318
  # bad data case
308
319
  recalibrated = []
309
- if result.bad_data:
320
+ if report.bad_data:
310
321
  logger.debug(
311
322
  f'"{workflow.__workflow_id__}": Bad data, diagnosing dependents')
312
323
  recalibrated = [
@@ -314,7 +325,7 @@ def diagnose(workflow: WorkflowType, code_path: str | Path,
314
325
  for n in get_dependents(workflow, code_path)
315
326
  ]
316
327
  if not any(recalibrated):
317
- if result.bad_data:
328
+ if report.bad_data:
318
329
  raise CalibrationFailedError(
319
330
  f'"{workflow.__workflow_id__}": bad data but no dependents recalibrated.'
320
331
  )
@@ -325,27 +336,27 @@ def diagnose(workflow: WorkflowType, code_path: str | Path,
325
336
  logger.debug(
326
337
  f'recalibrate "{workflow.__workflow_id__}" because some dependents recalibrated.'
327
338
  )
328
- elif not result.in_spec and not result.bad_data:
339
+ elif not report.in_spec and not report.bad_data:
329
340
  logger.debug(
330
341
  f'recalibrate "{workflow.__workflow_id__}" because out of spec.')
331
- elif result.in_spec:
342
+ elif report.in_spec:
332
343
  logger.error(
333
344
  f'Never reach: recalibrate "{workflow.__workflow_id__}" because in spec.'
334
345
  )
335
- elif result.bad_data:
346
+ elif report.bad_data:
336
347
  logger.error(
337
348
  f'Never reach: recalibrate "{workflow.__workflow_id__}" because bad data.'
338
349
  )
339
350
  else:
340
351
  logger.error(f'Never reach: recalibrate "{workflow.__workflow_id__}"')
341
352
 
342
- result = calibrate(workflow, code_path, state_path, plot, session_id)
343
- if result.bad_data or not result.in_spec:
344
- obey_the_oracle(result, state_path)
353
+ report = calibrate(workflow, code_path, state_path, plot, session_id)
354
+ if report.bad_data or not report.in_spec:
355
+ obey_the_oracle(report, state_path)
345
356
  raise CalibrationFailedError(
346
357
  f'"{workflow.__workflow_id__}": All dependents passed, but calibration failed!'
347
358
  )
348
- update_parameters(result, state_path)
359
+ update_parameters(report, state_path)
349
360
  return True
350
361
 
351
362
 
@@ -382,13 +393,13 @@ def maintain(workflow: WorkflowType,
382
393
  f'"{workflow.__workflow_id__}": In spec, no need to maintain')
383
394
  return
384
395
  # check_data
385
- result = check_data(workflow, code_path, state_path, plot, session_id)
386
- if result.in_spec:
396
+ report = check_data(workflow, code_path, state_path, plot, session_id)
397
+ if report.in_spec:
387
398
  if not run:
388
399
  logger.debug(
389
400
  f'"{workflow.__workflow_id__}": In spec, no need to maintain')
390
401
  return
391
- elif result.bad_data:
402
+ elif report.bad_data:
392
403
  logger.debug(
393
404
  f'"{workflow.__workflow_id__}": Bad data, diagnosing dependents')
394
405
  for n in get_dependents(workflow, code_path):
@@ -401,15 +412,15 @@ def maintain(workflow: WorkflowType,
401
412
  f'"{workflow.__workflow_id__}": All dependents diagnosed')
402
413
  # calibrate
403
414
  logger.debug(f'recalibrate "{workflow.__workflow_id__}"')
404
- result = calibrate(workflow, code_path, state_path, plot, session_id)
405
- if result.bad_data or not result.in_spec:
415
+ report = calibrate(workflow, code_path, state_path, plot, session_id)
416
+ if report.bad_data or not report.in_spec:
406
417
  if not freeze:
407
- obey_the_oracle(result, state_path)
418
+ obey_the_oracle(report, state_path)
408
419
  raise CalibrationFailedError(
409
420
  f'"{workflow.__workflow_id__}": All dependents passed, but calibration failed!'
410
421
  )
411
422
  if not freeze:
412
- update_parameters(result, state_path)
423
+ update_parameters(report, state_path)
413
424
  return
414
425
 
415
426
 
@@ -421,17 +432,17 @@ def run(workflow: WorkflowType,
421
432
  freeze: bool = False):
422
433
  session_id = uuid.uuid4().hex
423
434
  logger.debug(f'run "{workflow.__workflow_id__}" without dependences.')
424
- result = calibrate(workflow,
435
+ report = calibrate(workflow,
425
436
  code_path,
426
437
  state_path,
427
438
  plot,
428
439
  session_id=session_id)
429
- if result.bad_data or not result.in_spec:
440
+ if report.bad_data or not report.in_spec:
430
441
  if not freeze:
431
- obey_the_oracle(result, state_path)
442
+ obey_the_oracle(report, state_path)
432
443
  raise CalibrationFailedError(
433
444
  f'"{workflow.__workflow_id__}": All dependents passed, but calibration failed!'
434
445
  )
435
446
  if not freeze:
436
- update_parameters(result, state_path)
447
+ update_parameters(report, state_path)
437
448
  return
qulab/executor/storage.py CHANGED
@@ -16,7 +16,7 @@ __current_config_cache = None
16
16
 
17
17
 
18
18
  @dataclass
19
- class Result():
19
+ class Report():
20
20
  workflow: str = ''
21
21
  in_spec: bool = False
22
22
  bad_data: bool = False
@@ -38,7 +38,7 @@ class Result():
38
38
  @property
39
39
  def previous(self):
40
40
  if self.previous_path is not None and self.base_path is not None:
41
- return load_result(self.previous_path, self.base_path)
41
+ return load_report(self.previous_path, self.base_path)
42
42
  else:
43
43
  return None
44
44
 
@@ -89,7 +89,7 @@ def random_path(base: Path) -> Path:
89
89
  return path
90
90
 
91
91
 
92
- def save_config_key_history(key: str, result: Result,
92
+ def save_config_key_history(key: str, report: Report,
93
93
  base_path: str | Path) -> int:
94
94
  global __current_config_cache
95
95
  base_path = Path(base_path) / 'state'
@@ -103,14 +103,14 @@ def save_config_key_history(key: str, result: Result,
103
103
  __current_config_cache = {}
104
104
 
105
105
  __current_config_cache[
106
- key] = result.data, result.calibrated_time, result.checked_time
106
+ key] = report.data, report.calibrated_time, report.checked_time
107
107
 
108
108
  with open(base_path / 'parameters.pkl', 'wb') as f:
109
109
  pickle.dump(__current_config_cache, f)
110
110
  return 0
111
111
 
112
112
 
113
- def find_config_key_history(key: str, base_path: str | Path) -> Result | None:
113
+ def find_config_key_history(key: str, base_path: str | Path) -> Report | None:
114
114
  global __current_config_cache
115
115
  base_path = Path(base_path) / 'state'
116
116
  if __current_config_cache is None:
@@ -123,7 +123,7 @@ def find_config_key_history(key: str, base_path: str | Path) -> Result | None:
123
123
  if key in __current_config_cache:
124
124
  value, calibrated_time, checked_time = __current_config_cache.get(
125
125
  key, None)
126
- result = Result(
126
+ report = Report(
127
127
  workflow=f'cfg:{key}',
128
128
  bad_data=False,
129
129
  in_spec=True,
@@ -133,64 +133,65 @@ def find_config_key_history(key: str, base_path: str | Path) -> Result | None:
133
133
  calibrated_time=calibrated_time,
134
134
  checked_time=checked_time,
135
135
  )
136
- return result
136
+ return report
137
137
  return None
138
138
 
139
139
 
140
- def save_result(workflow: str,
141
- result: Result,
140
+ def save_report(workflow: str,
141
+ report: Report,
142
142
  base_path: str | Path,
143
- overwrite: bool = False) -> int:
143
+ overwrite: bool = False,
144
+ refresh_heads: bool = True) -> int:
144
145
  if workflow.startswith("cfg:"):
145
- return save_config_key_history(workflow[4:], result, base_path)
146
+ return save_config_key_history(workflow[4:], report, base_path)
146
147
 
147
148
  logger.debug(
148
- f'Saving result for "{workflow}", {result.in_spec=}, {result.bad_data=}, {result.fully_calibrated=}'
149
+ f'Saving report for "{workflow}", {report.in_spec=}, {report.bad_data=}, {report.fully_calibrated=}'
149
150
  )
150
151
  base_path = Path(base_path)
151
152
  if overwrite:
152
- buf = lzma.compress(pickle.dumps(result))
153
- path = result.path
153
+ buf = lzma.compress(pickle.dumps(report))
154
+ path = report.path
154
155
  if path is None:
155
- path = get_head(workflow, base_path)
156
- result.path = path
156
+ raise ValueError("Report path is None, can't overwrite.")
157
157
  with open(base_path / 'objects' / path, "rb") as f:
158
158
  index = int.from_bytes(f.read(8), 'big')
159
- result.index = index
159
+ report.index = index
160
160
  else:
161
- result.previous_path = get_head(workflow, base_path)
162
- buf = lzma.compress(pickle.dumps(result))
161
+ report.previous_path = get_head(workflow, base_path)
162
+ buf = lzma.compress(pickle.dumps(report))
163
163
  path = random_path(base_path / 'objects')
164
164
  (base_path / 'objects' / path).parent.mkdir(parents=True,
165
165
  exist_ok=True)
166
- result.path = path
167
- result.index = create_index("result",
166
+ report.path = path
167
+ report.index = create_index("report",
168
168
  base_path,
169
169
  context=str(path),
170
170
  width=35)
171
171
 
172
172
  with open(base_path / 'objects' / path, "wb") as f:
173
- f.write(result.index.to_bytes(8, 'big'))
173
+ f.write(report.index.to_bytes(8, 'big'))
174
174
  f.write(buf)
175
- set_head(workflow, path, base_path)
176
- return result.index
175
+ if refresh_heads:
176
+ set_head(workflow, path, base_path)
177
+ return report.index
177
178
 
178
179
 
179
- def load_result(path: str | Path, base_path: str | Path) -> Result | None:
180
+ def load_report(path: str | Path, base_path: str | Path) -> Report | None:
180
181
  base_path = Path(base_path)
181
182
  path = base_path / 'objects' / path
182
183
 
183
184
  with open(base_path / 'objects' / path, "rb") as f:
184
185
  index = int.from_bytes(f.read(8), 'big')
185
- result = pickle.loads(lzma.decompress(f.read()))
186
- result.base_path = base_path
187
- result.index = index
188
- return result
186
+ report = pickle.loads(lzma.decompress(f.read()))
187
+ report.base_path = base_path
188
+ report.index = index
189
+ return report
189
190
 
190
191
 
191
- def find_result(
192
+ def find_report(
192
193
  workflow: str, base_path: str | Path = get_config_value("data", Path)
193
- ) -> Result | None:
194
+ ) -> Report | None:
194
195
  if workflow.startswith("cfg:"):
195
196
  return find_config_key_history(workflow[4:], base_path)
196
197
 
@@ -198,25 +199,25 @@ def find_result(
198
199
  path = get_head(workflow, base_path)
199
200
  if path is None:
200
201
  return None
201
- return load_result(path, base_path)
202
+ return load_report(path, base_path)
202
203
 
203
204
 
204
- def renew_result(workflow: str, base_path: str | Path):
205
- logger.debug(f'Renewing result for "{workflow}"')
206
- result = find_result(workflow, base_path)
207
- if result is not None:
208
- result.checked_time = datetime.now()
209
- return save_result(workflow, result, base_path)
205
+ def renew_report(workflow: str, report, base_path: str | Path):
206
+ logger.debug(f'Renewing report for "{workflow}"')
207
+ report = find_report(workflow, base_path)
208
+ if report is not None:
209
+ report.checked_time = datetime.now()
210
+ return save_report(workflow, report, base_path)
210
211
 
211
212
 
212
- def revoke_result(workflow: str, base_path: str | Path):
213
- logger.debug(f'Revoking result for "{workflow}"')
213
+ def revoke_report(workflow: str, report, base_path: str | Path):
214
+ logger.debug(f'Revoking report for "{workflow}"')
214
215
  base_path = Path(base_path)
215
216
  path = get_head(workflow, base_path)
216
217
  if path is not None:
217
- result = load_result(path, base_path)
218
- result.in_spec = False
219
- return save_result(workflow, result, base_path)
218
+ report = load_report(path, base_path)
219
+ report.in_spec = False
220
+ return save_report(workflow, report, base_path)
220
221
 
221
222
 
222
223
  def set_head(workflow: str, path: Path, base_path: str | Path):
@@ -296,12 +297,12 @@ def query_index(name: str, base_path: str | Path, index: int):
296
297
  return context.rstrip()
297
298
 
298
299
 
299
- def get_result_by_index(
300
+ def get_report_by_index(
300
301
  index: int, base_path: str | Path = get_config_value("data", Path)
301
- ) -> Result | None:
302
+ ) -> Report | None:
302
303
  try:
303
- path = query_index("result", base_path, index)
304
- return load_result(path, base_path)
304
+ path = query_index("report", base_path, index)
305
+ return load_report(path, base_path)
305
306
  except:
306
307
  return None
307
308
 
@@ -1,4 +1,4 @@
1
- from .storage import Result, save_config
1
+ from .storage import Report, save_config
2
2
 
3
3
  __current_config_id = None
4
4
 
@@ -43,16 +43,16 @@ def _export_config() -> dict:
43
43
  return parameters
44
44
 
45
45
 
46
- def obey_the_oracle(result: Result, data_path):
46
+ def obey_the_oracle(report: Report, data_path):
47
47
  global __current_config_id
48
- update_config(result.oracle)
48
+ update_config(report.oracle)
49
49
  cfg = export_config()
50
50
  __current_config_id = save_config(cfg, data_path)
51
51
 
52
52
 
53
- def update_parameters(result: Result, data_path):
53
+ def update_parameters(report: Report, data_path):
54
54
  global __current_config_id
55
- update_config(result.parameters)
55
+ update_config(report.parameters)
56
56
  cfg = export_config()
57
57
  __current_config_id = save_config(cfg, data_path)
58
58
 
qulab/executor/utils.py CHANGED
@@ -37,7 +37,7 @@ def workflow_template(workflow: str, deps: list[str]) -> str:
37
37
  import numpy as np
38
38
  from loguru import logger
39
39
 
40
- from qulab.typing import Result
40
+ from qulab.typing import Report
41
41
 
42
42
 
43
43
  # 多长时间应该检查一次校准实验,单位是秒。
@@ -63,35 +63,35 @@ def calibrate():
63
63
  return x, y
64
64
 
65
65
 
66
- def analyze(result: Result, history: Result | None = None) -> Result:
66
+ def analyze(report: Report, history: Report | None = None) -> Report:
67
67
  \"\"\"
68
68
  分析校准结果。
69
69
 
70
- result: Result
70
+ report: Report
71
71
  本次校准实验的数据。
72
- history: Result | None
72
+ history: Report | None
73
73
  上次校准实验数据和分析结果,如果有的话。
74
74
  \"\"\"
75
75
  import random
76
76
 
77
- # 这里添加你的分析过程,运行 calibrate 得到的数据,在 result.data 里
77
+ # 这里添加你的分析过程,运行 calibrate 得到的数据,在 report.data 里
78
78
  # 你可以得到校准的结果,然后根据这个结果进行分析。
79
- x, y = result.data
79
+ x, y = report.data
80
80
 
81
81
  # 完整校准后的状态有两种:OK 和 Bad,分别对应校准成功和校准失败。
82
82
  # 校准失败是指出现坏数据,无法简单通过重新运行本次校准解决,需要
83
83
  # 检查前置步骤。
84
- result.state = random.choice(['OK', 'Bad'])
84
+ report.state = random.choice(['OK', 'Bad'])
85
85
 
86
86
  # 参数是一个字典,包含了本次校准得到的参数,后续会更新到config表中。
87
- result.parameters = {{'gate.R.Q1.params.amp':1}}
87
+ report.parameters = {{'gate.R.Q1.params.amp':1}}
88
88
 
89
89
  # 其他信息可以是任何可序列化的内容,你可以将你想要记录的信息放在这里。
90
90
  # 下次校准分析时,这些信息也会在 history 参数中一起传入,帮助你在下
91
91
  # 次分析时对比参考。
92
- result.other_infomation = {{}}
92
+ report.other_infomation = {{}}
93
93
 
94
- return result
94
+ return report
95
95
 
96
96
 
97
97
  def check():
@@ -111,33 +111,33 @@ def check():
111
111
  return x, y
112
112
 
113
113
 
114
- def check_analyze(result: Result, history: Result | None = None) -> Result:
114
+ def check_analyze(report: Report, history: Report | None = None) -> Report:
115
115
  \"\"\"
116
116
  分析检查结果。
117
117
 
118
- result: Result
118
+ report: Report
119
119
  本次检查实验的数据。
120
- history: Result | None
120
+ history: Report | None
121
121
  上次检查实验数据和分析结果,如果有的话。
122
122
  \"\"\"
123
123
  import random
124
124
 
125
- # 这里添加你的分析过程,运行 check 得到的数据,在 result.data 里
125
+ # 这里添加你的分析过程,运行 check 得到的数据,在 report.data 里
126
126
  # 你可以得到校准的结果,然后根据这个结果进行分析。
127
- x, y = result.data
127
+ x, y = report.data
128
128
 
129
129
  # 状态有三种:Outdated, OK, Bad,分别对应过时、正常、坏数据。
130
130
  # Outdated 是指数据过时,即参数漂了,需要重新校准。
131
131
  # OK 是指数据正常,参数也没漂,不用重新校准。
132
132
  # Bad 是指数据坏了,无法校准,需要检查前置步骤。
133
- result.state = random.choice(['Outdated', 'OK', 'Bad'])
133
+ report.state = random.choice(['Outdated', 'OK', 'Bad'])
134
134
 
135
- return result
135
+ return report
136
136
 
137
137
 
138
- def oracle(result: Result,
139
- history: Result | None = None,
140
- system_state: dict[str:str] | None = None) -> Result:
138
+ def oracle(report: Report,
139
+ history: Report | None = None,
140
+ system_state: dict[str:str] | None = None) -> Report:
141
141
  \"\"\"
142
142
  谕示:指凭直觉或经验判断,改动某些配置,以期望下次校准成功。
143
143
 
@@ -145,31 +145,31 @@ def oracle(result: Result,
145
145
  比如通常我们在死活测不到 rabi 或能谱时,会换一个 idle bias 再试试。这
146
146
  里我们凭直觉设的那个 bias 值,就是一个谕示,可以通过 oracle 来设定。
147
147
 
148
- 该函数代入的参数 result 是 analyze 函数的返回值。
148
+ 该函数代入的参数 report 是 analyze 函数的返回值。
149
149
  \"\"\"
150
150
 
151
- # result.oracle['Q0.bias'] = 0.1
152
- # result.oracle['Q1.bias'] = -0.03
151
+ # report.oracle['Q0.bias'] = 0.1
152
+ # report.oracle['Q1.bias'] = -0.03
153
153
 
154
- return result
154
+ return report
155
155
  """
156
156
 
157
157
 
158
158
  def debug_analyze(
159
- result_index: int,
159
+ report_index: int,
160
160
  code_path: str | Path = get_config_value('code', Path),
161
161
  data_path: str | Path = get_config_value('data', Path),
162
162
  ) -> None:
163
- from .storage import get_result_by_index
163
+ from .storage import get_report_by_index
164
164
 
165
- result = get_result_by_index(result_index, data_path)
166
- if result is None:
167
- raise ValueError(f'Invalid result index: {result_index}')
168
- workflow = result.workflow
165
+ report = get_report_by_index(report_index, data_path)
166
+ if report is None:
167
+ raise ValueError(f'Invalid report index: {report_index}')
168
+ workflow = report.workflow
169
169
  wf = load_workflow(workflow, code_path)
170
170
  if wf is None:
171
171
  raise ValueError(f'Invalid workflow: {workflow}')
172
- result = wf.analyze(result, result.previous)
172
+ report = wf.analyze(report, report.previous)
173
173
  if hasattr(wf, 'plot'):
174
- wf.plot(result)
175
- return result
174
+ wf.plot(report)
175
+ return report
Binary file
qulab/typing.py CHANGED
@@ -1,2 +1,2 @@
1
- from .executor.storage import Result
1
+ from .executor.storage import Report
2
2
  from .scan.record import Record
qulab/version.py CHANGED
@@ -1 +1 @@
1
- __version__ = "2.7.6"
1
+ __version__ = "2.7.7"
File without changes
File without changes