QuLab 2.4.16__cp311-cp311-win_amd64.whl → 2.4.18__cp311-cp311-win_amd64.whl

{QuLab-2.4.16.dist-info → QuLab-2.4.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: QuLab
-Version: 2.4.16
+Version: 2.4.18
 Summary: contral instruments and manage data
 Author-email: feihoo87 <feihoo87@gmail.com>
 Maintainer-email: feihoo87 <feihoo87@gmail.com>

{QuLab-2.4.16.dist-info → QuLab-2.4.18.dist-info}/RECORD

@@ -1,17 +1,17 @@
 qulab/__init__.py,sha256=vkFybY8YSsQilYdThPRD83-btPAR41sy_WCXiM-6mME,141
 qulab/__main__.py,sha256=g9iBs8xxX6Yik7cmgllQkpBN8C4JNoZVsEOyCCLCyFU,63
 qulab/dicttree.py,sha256=ZoSJVWK4VMqfzj42gPb_n5RqLlM6K1Me0WmLIfLEYf8,14195
-qulab/fun.cp311-win_amd64.pyd,sha256=Ud9frCe9jfVXvl_IfrEf97TZotrnIWFyOn7gHeDeYrM,31744
-qulab/version.py,sha256=rVOnxEJVvl0ZWvCblBzVBGcp0VALbV_koS5W-UB9zfk,22
+qulab/fun.cp311-win_amd64.pyd,sha256=YzDNJ5JsetL6lqjZ0ZOcdNYGJP03vyX4pvWSG93s3mw,31744
+qulab/version.py,sha256=vQo2edKgYGHn5FfacX8TS2Blt-3b9unuA9CvJvhZhfo,22
 qulab/cli/__init__.py,sha256=6xd2eYw32k1NmfAuYSu__1kaP12Oz1QVqwbkYXdWno4,588
 qulab/cli/config.py,sha256=gvMObzaVJ-xTJ1GmhP3sATayfDasYlijX1krSGeccNY,3071
 qulab/executor/__init__.py,sha256=LosPzOMaljSZY1thy_Fxtbrgq7uubJszMABEB7oM7tU,101
-qulab/executor/cli.py,sha256=LRSb5-J8Tn7dj-fxwCp3OP_6YUgPRoY-f8O18sbWG8I,5164
-qulab/executor/load.py,sha256=yTLoWOtTvmCTuYFZ6qFsjBM_CvsrKgmUkBSHHz3223A,10899
-qulab/executor/schedule.py,sha256=SfMZys8WdmIvfW5fPM42g-F6XwewTdpC0-puhf5TA54,11721
+qulab/executor/cli.py,sha256=owdDSaSuzTPQA5SlNCxX4nHWITU1BF0VZMj7tLqCZhM,5781
+qulab/executor/load.py,sha256=eKysHL5EHKd_YcfHOVwRoaNsD-k0cTuxDY_7Yn5W8hA,11078
+qulab/executor/schedule.py,sha256=bXQzn3e3tCvAVdEBsvh5ecAThP52njyxbsrlrxC9TzA,11922
 qulab/executor/storage.py,sha256=M66Q5_Uc5MMfc_QAuuaaexwAz7wxBPMkeleB5nRpQmI,4621
 qulab/executor/transform.py,sha256=inaOn6eqCs22ZZ0xAQl8s8YCoEACaXSwFNNu7jqdwAk,2148
-qulab/executor/utils.py,sha256=jp3F0swO9gpKhmTauxbgEK9WVorjI9FAL6zT60cYTdk,3106
+qulab/executor/utils.py,sha256=RmnT0wMTAvavM-abgFt2tQN8TaaHuk86-YAiOUQCoBU,3101
 qulab/monitor/__init__.py,sha256=xEVDkJF8issrsDeLqQmDsvtRmrf-UiViFcGTWuzdlFU,43
 qulab/monitor/__main__.py,sha256=k2H1H5Zf9LLXTDLISJkbikLH-z0f1e5i5i6wXXYPOrE,105
 qulab/monitor/config.py,sha256=y_5StMkdrbZO1ziyKBrvIkB7Jclp9RCPK1QbsOhCxnY,785
@@ -91,9 +91,9 @@ qulab/visualization/plot_seq.py,sha256=Uo1-dB1YE9IN_A9tuaOs9ZG3S5dKDQ_l98iD2Wbxp
 qulab/visualization/qdat.py,sha256=HubXFu4nfcA7iUzghJGle1C86G6221hicLR0b-GqhKQ,5887
 qulab/visualization/rot3d.py,sha256=jGHJcqj1lEWBUV-W4GUGONGacqjrYvuFoFCwPse5h1Y,757
 qulab/visualization/widgets.py,sha256=HcYwdhDtLreJiYaZuN3LfofjJmZcLwjMfP5aasebgDo,3266
-QuLab-2.4.16.dist-info/LICENSE,sha256=b4NRQ-GFVpJMT7RuExW3NwhfbrYsX7AcdB7Gudok-fs,1086
-QuLab-2.4.16.dist-info/METADATA,sha256=800PogLCuT2uN9w6hfEuVv5sOrzXlr3AobBHeiruLrQ,3804
-QuLab-2.4.16.dist-info/WHEEL,sha256=yNnHoQL2GZYIUXm9YvoaBpFjGlUoK9qq9oqYeudrWlE,101
-QuLab-2.4.16.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
-QuLab-2.4.16.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
-QuLab-2.4.16.dist-info/RECORD,,
+QuLab-2.4.18.dist-info/LICENSE,sha256=b4NRQ-GFVpJMT7RuExW3NwhfbrYsX7AcdB7Gudok-fs,1086
+QuLab-2.4.18.dist-info/METADATA,sha256=-FnFnk9TfHLKH7FWUVaR1TrbMPpjKBCyN2D7FSnUECg,3804
+QuLab-2.4.18.dist-info/WHEEL,sha256=yNnHoQL2GZYIUXm9YvoaBpFjGlUoK9qq9oqYeudrWlE,101
+QuLab-2.4.18.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
+QuLab-2.4.18.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
+QuLab-2.4.18.dist-info/RECORD,,

qulab/executor/cli.py

@@ -54,6 +54,7 @@ def create(workflow, code):
     """
     Create a new workflow file.
     """
+    logger.info(f'[CMD]: create {workflow} --code {code}')
     if code is None:
         code = Path.cwd()
 
@@ -82,6 +83,7 @@ def set(key, value, api):
     """
     Set a config.
     """
+    logger.info(f'[CMD]: set {key} {value} --api {api}')
     from . import transform
     if api is not None:
         api = importlib.import_module(api)
@@ -103,6 +105,7 @@ def get(key, api):
     """
     Get a config.
     """
+    logger.info(f'[CMD]: get {key} --api {api}')
     from . import transform
     if api is not None:
         api = importlib.import_module(api)
@@ -117,12 +120,16 @@ def get(key, api):
               '-n',
               is_flag=True,
               help='Do not run dependents.')
+@click.option('--update', '-u', is_flag=True)
 @log_options
 @command_option('run')
-def run(workflow, code, data, api, plot, no_dependents):
+def run(workflow, code, data, api, plot, no_dependents, update):
     """
     Run a workflow.
     """
+    logger.info(
+        f'[CMD]: run {workflow} --code {code} --data {data} --api {api}{" --plot" if plot else ""}{" --no-dependents" if no_dependents else ""}{" --update " if update else ""}'
+    )
     if api is not None:
         api = importlib.import_module(api)
         set_config_api(api.query_config, api.update_config)
@@ -139,15 +146,15 @@ def run(workflow, code, data, api, plot, no_dependents):
     if no_dependents:
         if hasattr(wf, 'entries'):
             for entry in get_entries(wf, code):
-                run_workflow(entry, code, data, plot=plot)
+                run_workflow(entry, code, data, plot=plot, update=update)
         else:
-            run_workflow(wf, code, data, plot=plot)
+            run_workflow(wf, code, data, plot=plot, update=update)
     else:
         if hasattr(wf, 'entries'):
             for entry in get_entries(wf, code):
-                maintain_workflow(entry, code, data, run=True, plot=plot)
+                maintain_workflow(entry, code, data, run=True, plot=plot, update=update)
         else:
-            maintain_workflow(wf, code, data, run=True, plot=plot)
+            maintain_workflow(wf, code, data, run=True, plot=plot, update=update)
 
 
 @click.command()
@@ -159,6 +166,9 @@ def maintain(workflow, code, data, api, plot):
     """
     Maintain a workflow.
     """
+    logger.info(
+        f'[CMD]: maintain {workflow} --code {code} --data {data} --api {api}{" --plot" if plot else ""}'
+    )
     if api is not None:
         api = importlib.import_module(api)
         set_config_api(api.query_config, api.update_config)

qulab/executor/load.py

@@ -21,7 +21,7 @@ class SetConfigWorkflow():
         self.key = key
 
     def depends(self):
-        return [[]]
+        return []
 
     def check_state(self, history: Result) -> bool:
         from . import transform
@@ -254,6 +254,10 @@ def load_workflow_from_template(template_path: str,
         return new_text
 
     template = string.Template(replace(content))
+    keys = template.get_identifiers()
+    missing = set(keys) - set(mappping.keys())
+    if missing:
+        raise KeyError(f"{template_path}: Missing keys in mapping: {missing}")
     content = template.substitute(mappping)
 
     hash_str = hashlib.md5(pickle.dumps(mappping)).hexdigest()[:8]
@@ -323,7 +327,7 @@ def get_dependents(workflow: WorkflowType,
                    code_path: str | Path) -> list[WorkflowType]:
     return [
         load_workflow(n, code_path, mtime=workflow.__mtime__)
-        for n in workflow.depends()[0]
+        for n in workflow.depends()
     ]
 
 

qulab/executor/schedule.py

@@ -236,7 +236,8 @@ def maintain(workflow: WorkflowType,
              state_path: str | Path,
              session_id: str | None = None,
              run: bool = False,
-             plot: bool = False):
+             plot: bool = False,
+             update: bool = True):
     if session_id is None:
         session_id = uuid.uuid4().hex
     logger.debug(f'run "{workflow.__workflow_id__}"'
@@ -280,7 +281,8 @@ def maintain(workflow: WorkflowType,
         raise CalibrationFailedError(
             f'"{workflow.__workflow_id__}": All dependents passed, but calibration failed!'
         )
-    transform.update_parameters(result)
+    if update:
+        transform.update_parameters(result)
     return
 
 
@@ -288,13 +290,19 @@ def maintain(workflow: WorkflowType,
 def run(workflow: WorkflowType,
         code_path: str | Path,
         state_path: str | Path,
-        plot: bool = False):
+        plot: bool = False,
+        update: bool = True):
     session_id = uuid.uuid4().hex
     logger.debug(f'run "{workflow.__workflow_id__}" without dependences.')
-    result = calibrate(workflow, code_path, state_path, plot, session_id=session_id)
+    result = calibrate(workflow,
+                       code_path,
+                       state_path,
+                       plot,
+                       session_id=session_id)
     if result.bad_data or not result.in_spec:
         raise CalibrationFailedError(
             f'"{workflow.__workflow_id__}": All dependents passed, but calibration failed!'
         )
-    transform.update_parameters(result)
+    if update:
+        transform.update_parameters(result)
     return

qulab/executor/utils.py

@@ -25,7 +25,7 @@ def dependent_tree(node: str, code_path: str | Path) -> dict[str, list[str]]:
     Returns a dict of nodes and their dependents.
     '''
     tree = {}
-    for n in load_workflow(node, code_path).depends()[0]:
+    for n in load_workflow(node, code_path).depends():
         tree[n] = dependent_tree(n, code_path)
     return tree
 
@@ -43,7 +43,7 @@ import numpy as np
 __timeout__ = 7*24*3600
 
 def depends():
-    return [{deps!r}]
+    return {deps!r}
 
 
 def calibrate():

qulab/version.py

@@ -1 +1 @@
-__version__ = "2.4.16"
+__version__ = "2.4.18"