QuLab 2.7.6__cp310-cp310-macosx_10_9_universal2.whl → 2.7.8__cp310-cp310-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.8
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-310-darwin.so,sha256=QgEHTFx4ediOVIUKGllGdaak8zXnkawvicMAoWGO33E,126864
5
- qulab/typing.py,sha256=5xCLfrp1aZpKpDy4p2arbFszw2eK3hGUjZa-XSvC_-8,69
4
+ qulab/fun.cpython-310-darwin.so,sha256=e1CeXKgrro8pI7OqXpBYdixor8uqX1GwkCkuCc1sQSs,126864
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=89JN-VCK3x9bbDL4mH6FsYHIsGSTeoIxhxxTqfViSLE,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=OYvolWiafZbXiodeyJvr0-hscHvBt-eifmg7uWEymII,16952
15
+ qulab/executor/storage.py,sha256=JH817-d6izBiz6wBGWvWxNEY_IPhRwRmGcB_eUpvE8U,11364
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=Yd3eJSBM2hj8W-ouaiMfFUwQYAS-D6P73Ob9yN5MZd0,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.8.dist-info/LICENSE,sha256=PRzIKxZtpQcH7whTG6Egvzl1A0BvnSf30tmR2X2KrpA,1065
98
+ QuLab-2.7.8.dist-info/METADATA,sha256=ZZQMlhagxeJHogg-wXRxB2SaFziTmaBjXwy05kW4fps,3698
99
+ QuLab-2.7.8.dist-info/WHEEL,sha256=Yd3eJSBM2hj8W-ouaiMfFUwQYAS-D6P73Ob9yN5MZd0,114
100
+ QuLab-2.7.8.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
101
+ QuLab-2.7.8.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
102
+ QuLab-2.7.8.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_head, 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,134 @@ 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,
126
- check=False,
127
- plot=False) -> Result:
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
123
+ @logger.catch()
124
+ def call_plot(node: WorkflowType, report: Report, check=False):
125
+ if hasattr(node, 'plot') and callable(node.plot):
126
+ node.plot(report)
127
+
128
+
129
+ def call_check(workflow: WorkflowType, session_id: str, state_path: Path):
130
+ report = get_cache(session_id, (workflow, 'check'))
131
+ if report is not None:
132
+ logger.debug(f'Cache hit for "{workflow.__workflow_id__}:check"')
133
+ return report
134
+
135
+ data = workflow.check()
136
+ if not is_pickleable(data):
137
+ raise TypeError(
138
+ f'"{workflow.__workflow_id__}" : "check" return not pickleable data'
139
+ )
140
+ report = Report(workflow=workflow.__workflow_id__,
141
+ data=data,
142
+ config_path=current_config(state_path),
143
+ base_path=state_path,
144
+ heads=get_heads(state_path),
145
+ previous_path=get_head(workflow.__workflow_id__,
146
+ state_path))
147
+
148
+ save_report(workflow.__workflow_id__, report, state_path)
149
+
150
+ set_cache(session_id, (workflow, 'check'), report)
151
+ return report
152
+
153
+
154
+ def call_calibrate(workflow: WorkflowType, session_id: str, state_path: Path):
155
+ report = get_cache(session_id, (workflow, 'calibrate'))
156
+ if report is not None:
157
+ logger.debug(f'Cache hit for "{workflow.__workflow_id__}:calibrate"')
158
+ return report
159
+
160
+ data = workflow.calibrate()
161
+ if not is_pickleable(data):
162
+ raise TypeError(
163
+ f'"{workflow.__workflow_id__}" : "calibrate" return not pickleable data'
164
+ )
165
+ report = Report(workflow=workflow.__workflow_id__,
166
+ data=data,
167
+ config_path=current_config(state_path),
168
+ base_path=state_path,
169
+ heads=get_heads(state_path),
170
+ previous_path=get_head(workflow.__workflow_id__,
171
+ state_path))
172
+
173
+ save_report(workflow.__workflow_id__, report, state_path)
174
+
175
+ set_cache(session_id, (workflow, 'calibrate'), report)
176
+ return report
177
+
178
+
179
+ def call_check_analyzer(node: WorkflowType,
180
+ report: Report,
181
+ history: Report | None,
182
+ state_path: Path,
183
+ plot=False) -> Report:
184
+ report = node.check_analyze(report, history=history)
185
+ veryfy_analyzed_report(report, node.__workflow_id__, "check_analyze")
186
+ report.fully_calibrated = False
187
+ if report.in_spec:
188
+ logger.debug(
189
+ f'"{node.__workflow_id__}": checked in spec, renewing report')
190
+ renew_report(node.__workflow_id__, report.previous, state_path)
132
191
  else:
133
- result = node.analyze(result, history=history)
134
- veryfy_analyzed_result(result, node.__workflow_id__, "analyze")
135
- if hasattr(node, 'oracle') and callable(node.oracle):
136
- logger.debug(
137
- f'"{node.__workflow_id__}" has oracle method, calling ...')
138
- try:
139
- result = node.oracle(result,
140
- history=history,
141
- system_state=get_heads(result.base_path))
142
- except Exception as e:
143
- logger.exception(e)
144
- result.oracle = {}
145
- if not isinstance(result, Result):
146
- raise TypeError(
147
- f'"{node.__workflow_id__}" : function "oracle" must return a Result object'
148
- )
149
- if not is_pickleable(result.oracle):
150
- raise TypeError(
151
- f'"{node.__workflow_id__}" : function "oracle" return not pickleable data'
152
- )
153
- result.fully_calibrated = True
154
- if plot:
155
- call_plot(node, result)
156
- return result
192
+ logger.debug(
193
+ f'"{node.__workflow_id__}": checked out of spec, revoking report')
194
+ revoke_report(node.__workflow_id__, report.previous, state_path)
195
+ return report
157
196
 
158
197
 
159
- @logger.catch()
160
- def call_plot(node, result: Result, check=False):
161
- if hasattr(node, 'plot') and callable(node.plot):
162
- node.plot(result)
198
+ def call_analyzer(node: WorkflowType,
199
+ report: Report,
200
+ history: Report | None,
201
+ state_path: Path,
202
+ plot=False) -> Report:
203
+
204
+ report = node.analyze(report, history=history)
205
+ veryfy_analyzed_report(report, node.__workflow_id__, "analyze")
206
+ if hasattr(node, 'oracle') and callable(node.oracle):
207
+ logger.debug(
208
+ f'"{node.__workflow_id__}" has oracle method, calling ...')
209
+ try:
210
+ report = node.oracle(report,
211
+ history=history,
212
+ system_state=get_heads(report.base_path))
213
+ except Exception as e:
214
+ logger.exception(e)
215
+ report.oracle = {}
216
+ if not isinstance(report, Report):
217
+ raise TypeError(
218
+ f'"{node.__workflow_id__}" : function "oracle" must return a Report object'
219
+ )
220
+ if not is_pickleable(report.oracle):
221
+ raise TypeError(
222
+ f'"{node.__workflow_id__}" : function "oracle" return not pickleable data'
223
+ )
224
+ report.fully_calibrated = True
225
+ save_report(node.__workflow_id__, report, state_path, overwrite=True)
226
+ if plot:
227
+ call_plot(node, report)
228
+ return report
163
229
 
164
230
 
165
- def check_data(workflow: WorkflowType, code_path: str | Path,
166
- state_path: str | Path, plot: bool, session_id: str) -> Result:
231
+ def check_data(workflow: WorkflowType, state_path: str | Path, plot: bool,
232
+ session_id: str) -> Report:
167
233
  """
168
234
  check data answers two questions:
169
235
  Is the parameter associated with this cal in spec,
170
236
  and is the cal scan working as expected?
171
237
  """
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)
238
+ history = find_report(workflow.__workflow_id__, state_path)
178
239
 
179
240
  if history is None:
180
241
  logger.debug(f'No history found for "{workflow.__workflow_id__}"')
181
- result = Result(workflow=workflow.__workflow_id__,
242
+ report = Report(workflow=workflow.__workflow_id__,
182
243
  config_path=current_config(state_path),
183
244
  base_path=state_path,
184
- heads=get_heads(state_path))
185
- result.in_spec = False
186
- result.bad_data = False
187
- return result
245
+ heads=get_heads(state_path),
246
+ previous_path=get_head(workflow.__workflow_id__,
247
+ state_path))
248
+ report.in_spec = False
249
+ report.bad_data = False
250
+ return report
188
251
 
189
252
  if history.bad_data:
190
253
  logger.debug(
@@ -202,92 +265,48 @@ def check_data(workflow: WorkflowType, code_path: str | Path,
202
265
  workflow, 'check_analyze') and callable(workflow.check_analyze):
203
266
  logger.debug(
204
267
  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)
268
+
269
+ report = call_check(workflow, session_id, state_path)
216
270
 
217
271
  logger.debug(f'Checked "{workflow.__workflow_id__}" !')
218
- result = call_analyzer(workflow,
219
- result,
220
- history,
221
- check=True,
222
- plot=plot)
223
- if result.in_spec:
224
- logger.debug(
225
- f'"{workflow.__workflow_id__}": checked in spec, renewing result'
226
- )
227
- renew_result(workflow.__workflow_id__, state_path)
228
- else:
229
- logger.debug(
230
- f'"{workflow.__workflow_id__}": checked out of spec, revoking result'
231
- )
232
- revoke_result(workflow.__workflow_id__, state_path)
272
+ report = call_check_analyzer(workflow,
273
+ report,
274
+ history,
275
+ state_path,
276
+ plot=plot)
233
277
  else:
234
278
  logger.debug(
235
279
  f'Checking "{workflow.__workflow_id__}" with "calibrate" method ...'
236
280
  )
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)
281
+
282
+ report = call_calibrate(workflow, session_id, state_path)
248
283
 
249
284
  logger.debug(f'Calibrated "{workflow.__workflow_id__}" !')
250
- result = call_analyzer(workflow,
251
- result,
285
+ report = call_analyzer(workflow,
286
+ report,
252
287
  history,
253
- check=False,
288
+ state_path,
254
289
  plot=plot)
255
- save_result(workflow.__workflow_id__,
256
- result,
257
- state_path,
258
- overwrite=True)
290
+ return report
259
291
 
260
- set_cache(session_id, (workflow, 'check_data'), result)
261
- return result
262
292
 
293
+ def calibrate(workflow: WorkflowType, state_path: str | Path, plot: bool,
294
+ session_id: str) -> Report:
295
+ history = find_report(workflow.__workflow_id__, state_path)
263
296
 
264
- 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
297
+ logger.debug(f'Calibrating "{workflow.__workflow_id__}" ...')
270
298
 
271
- history = find_result(workflow.__workflow_id__, state_path)
299
+ report = call_calibrate(workflow, session_id, state_path)
272
300
 
273
- 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)
285
301
  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
302
 
289
- set_cache(session_id, (workflow, 'calibrate'), result)
290
- return result
303
+ report = call_analyzer(workflow,
304
+ report,
305
+ history,
306
+ state_path,
307
+ check=False,
308
+ plot=plot)
309
+ return report
291
310
 
292
311
 
293
312
  def diagnose(workflow: WorkflowType, code_path: str | Path,
@@ -297,16 +316,16 @@ def diagnose(workflow: WorkflowType, code_path: str | Path,
297
316
  '''
298
317
  logger.debug(f'diagnose "{workflow.__workflow_id__}"')
299
318
  # check_data
300
- result = check_data(workflow, code_path, state_path, plot, session_id)
319
+ report = check_data(workflow, state_path, plot, session_id)
301
320
  # in spec case
302
- if result.in_spec:
321
+ if report.in_spec:
303
322
  logger.debug(
304
323
  f'"{workflow.__workflow_id__}": Checked! In spec, no need to diagnose'
305
324
  )
306
325
  return False
307
326
  # bad data case
308
327
  recalibrated = []
309
- if result.bad_data:
328
+ if report.bad_data:
310
329
  logger.debug(
311
330
  f'"{workflow.__workflow_id__}": Bad data, diagnosing dependents')
312
331
  recalibrated = [
@@ -314,7 +333,7 @@ def diagnose(workflow: WorkflowType, code_path: str | Path,
314
333
  for n in get_dependents(workflow, code_path)
315
334
  ]
316
335
  if not any(recalibrated):
317
- if result.bad_data:
336
+ if report.bad_data:
318
337
  raise CalibrationFailedError(
319
338
  f'"{workflow.__workflow_id__}": bad data but no dependents recalibrated.'
320
339
  )
@@ -325,27 +344,27 @@ def diagnose(workflow: WorkflowType, code_path: str | Path,
325
344
  logger.debug(
326
345
  f'recalibrate "{workflow.__workflow_id__}" because some dependents recalibrated.'
327
346
  )
328
- elif not result.in_spec and not result.bad_data:
347
+ elif not report.in_spec and not report.bad_data:
329
348
  logger.debug(
330
349
  f'recalibrate "{workflow.__workflow_id__}" because out of spec.')
331
- elif result.in_spec:
350
+ elif report.in_spec:
332
351
  logger.error(
333
352
  f'Never reach: recalibrate "{workflow.__workflow_id__}" because in spec.'
334
353
  )
335
- elif result.bad_data:
354
+ elif report.bad_data:
336
355
  logger.error(
337
356
  f'Never reach: recalibrate "{workflow.__workflow_id__}" because bad data.'
338
357
  )
339
358
  else:
340
359
  logger.error(f'Never reach: recalibrate "{workflow.__workflow_id__}"')
341
360
 
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)
361
+ report = calibrate(workflow, state_path, plot, session_id)
362
+ if report.bad_data or not report.in_spec:
363
+ obey_the_oracle(report, state_path)
345
364
  raise CalibrationFailedError(
346
365
  f'"{workflow.__workflow_id__}": All dependents passed, but calibration failed!'
347
366
  )
348
- update_parameters(result, state_path)
367
+ update_parameters(report, state_path)
349
368
  return True
350
369
 
351
370
 
@@ -382,13 +401,13 @@ def maintain(workflow: WorkflowType,
382
401
  f'"{workflow.__workflow_id__}": In spec, no need to maintain')
383
402
  return
384
403
  # check_data
385
- result = check_data(workflow, code_path, state_path, plot, session_id)
386
- if result.in_spec:
404
+ report = check_data(workflow, state_path, plot, session_id)
405
+ if report.in_spec:
387
406
  if not run:
388
407
  logger.debug(
389
408
  f'"{workflow.__workflow_id__}": In spec, no need to maintain')
390
409
  return
391
- elif result.bad_data:
410
+ elif report.bad_data:
392
411
  logger.debug(
393
412
  f'"{workflow.__workflow_id__}": Bad data, diagnosing dependents')
394
413
  for n in get_dependents(workflow, code_path):
@@ -401,15 +420,15 @@ def maintain(workflow: WorkflowType,
401
420
  f'"{workflow.__workflow_id__}": All dependents diagnosed')
402
421
  # calibrate
403
422
  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:
423
+ report = calibrate(workflow, state_path, plot, session_id)
424
+ if report.bad_data or not report.in_spec:
406
425
  if not freeze:
407
- obey_the_oracle(result, state_path)
426
+ obey_the_oracle(report, state_path)
408
427
  raise CalibrationFailedError(
409
428
  f'"{workflow.__workflow_id__}": All dependents passed, but calibration failed!'
410
429
  )
411
430
  if not freeze:
412
- update_parameters(result, state_path)
431
+ update_parameters(report, state_path)
413
432
  return
414
433
 
415
434
 
@@ -421,17 +440,13 @@ def run(workflow: WorkflowType,
421
440
  freeze: bool = False):
422
441
  session_id = uuid.uuid4().hex
423
442
  logger.debug(f'run "{workflow.__workflow_id__}" without dependences.')
424
- result = calibrate(workflow,
425
- code_path,
426
- state_path,
427
- plot,
428
- session_id=session_id)
429
- if result.bad_data or not result.in_spec:
443
+ report = calibrate(workflow, state_path, plot, session_id=session_id)
444
+ if report.bad_data or not report.in_spec:
430
445
  if not freeze:
431
- obey_the_oracle(result, state_path)
446
+ obey_the_oracle(report, state_path)
432
447
  raise CalibrationFailedError(
433
448
  f'"{workflow.__workflow_id__}": All dependents passed, but calibration failed!'
434
449
  )
435
450
  if not freeze:
436
- update_parameters(result, state_path)
451
+ update_parameters(report, state_path)
437
452
  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)
152
+ try:
153
+ buf = lzma.compress(pickle.dumps(report))
154
+ except:
155
+ raise ValueError(f"Can't pickle report for {workflow}")
151
156
  if overwrite:
152
- buf = lzma.compress(pickle.dumps(result))
153
- path = result.path
157
+ path = report.path
154
158
  if path is None:
155
- path = get_head(workflow, base_path)
156
- result.path = path
159
+ raise ValueError("Report path is None, can't overwrite.")
157
160
  with open(base_path / 'objects' / path, "rb") as f:
158
161
  index = int.from_bytes(f.read(8), 'big')
159
- result.index = index
162
+ report.index = index
160
163
  else:
161
- result.previous_path = get_head(workflow, base_path)
162
- buf = lzma.compress(pickle.dumps(result))
163
164
  path = random_path(base_path / 'objects')
164
165
  (base_path / 'objects' / path).parent.mkdir(parents=True,
165
166
  exist_ok=True)
166
- result.path = path
167
- result.index = create_index("result",
167
+ report.path = path
168
+ report.index = create_index("report",
168
169
  base_path,
169
170
  context=str(path),
170
171
  width=35)
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,33 @@ 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
-
203
-
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)
202
+ return load_report(path, base_path)
203
+
204
+
205
+ def renew_report(workflow: str, report: Report | None, base_path: str | Path):
206
+ logger.debug(f'Renewing report for "{workflow}"')
207
+ if report is not None:
208
+ report.checked_time = datetime.now()
209
+ return save_report(workflow,
210
+ report,
211
+ base_path,
212
+ overwrite=True,
213
+ refresh_heads=True)
214
+ else:
215
+ raise ValueError(f"Can't renew report for {workflow}")
210
216
 
211
217
 
212
- def revoke_result(workflow: str, base_path: str | Path):
213
- logger.debug(f'Revoking result for "{workflow}"')
218
+ def revoke_report(workflow: str, report: Report | None, base_path: str | Path):
219
+ logger.debug(f'Revoking report for "{workflow}"')
214
220
  base_path = Path(base_path)
215
- path = get_head(workflow, base_path)
216
- 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)
221
+ if report is not None:
222
+ report.in_spec = False
223
+ report.previous_path = report.path
224
+ return save_report(workflow,
225
+ report,
226
+ base_path,
227
+ overwrite=False,
228
+ refresh_heads=True)
220
229
 
221
230
 
222
231
  def set_head(workflow: str, path: Path, base_path: str | Path):
@@ -296,12 +305,12 @@ def query_index(name: str, base_path: str | Path, index: int):
296
305
  return context.rstrip()
297
306
 
298
307
 
299
- def get_result_by_index(
308
+ def get_report_by_index(
300
309
  index: int, base_path: str | Path = get_config_value("data", Path)
301
- ) -> Result | None:
310
+ ) -> Report | None:
302
311
  try:
303
- path = query_index("result", base_path, index)
304
- return load_result(path, base_path)
312
+ path = query_index("report", base_path, index)
313
+ return load_report(path, base_path)
305
314
  except:
306
315
  return None
307
316
 
@@ -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.8"
File without changes
File without changes