QuLab 2.9.6__cp310-cp310-macosx_10_9_universal2.whl → 2.9.7__cp310-cp310-macosx_10_9_universal2.whl

qulab/executor/cli.py

@@ -126,7 +126,8 @@ def set(key, value, api):
     from . import transform
     if api is not None:
         api = importlib.import_module(api)
-        set_config_api(api.query_config, api.update_config, api.export_config)
+        set_config_api(api.query_config, api.update_config, api.delete_config,
+                       api.export_config, api.clear_config)
     try:
         value = eval(value)
     except:
@@ -148,7 +149,8 @@ def get(key, api):
     from . import transform
     if api is not None:
         api = importlib.import_module(api)
-        set_config_api(api.query_config, api.update_config, api.export_config)
+        set_config_api(api.query_config, api.update_config, api.delete_config,
+                       api.export_config, api.clear_config)
     click.echo(transform.query_config(key))
 
 
@@ -184,7 +186,8 @@ def run(workflow, code, data, api, plot, no_dependents, retry, freeze):
         f'{" --freeze " if freeze else ""}')
     if api is not None:
         api = importlib.import_module(api)
-        set_config_api(api.query_config, api.update_config, api.export_config)
+        set_config_api(api.query_config, api.update_config, api.delete_config,
+                       api.export_config, api.clear_config)
     if code is None:
         code = Path.cwd()
     if data is None:
@@ -255,7 +258,8 @@ def maintain(workflow, code, data, api, retry, plot):
         f'{" --plot" if plot else ""}')
     if api is not None:
         api = importlib.import_module(api)
-        set_config_api(api.query_config, api.update_config, api.export_config)
+        set_config_api(api.query_config, api.update_config, api.delete_config,
+                       api.export_config, api.clear_config)
     if code is None:
         code = Path.cwd()
     if data is None:
@@ -311,7 +315,8 @@ def reproduce(report_id, code, data, api, plot):
         f'{" --plot" if plot else ""}')
     if api is not None:
         api = importlib.import_module(api)
-        set_config_api(api.query_config, api.update_config, api.export_config)
+        set_config_api(api.query_config, api.update_config, api.delete_config,
+                       api.export_config, api.clear_config)
     if code is None:
         code = Path.cwd()
     if data is None:
@@ -320,10 +325,16 @@ def reproduce(report_id, code, data, api, plot):
     code = Path(os.path.expanduser(code))
     data = Path(os.path.expanduser(data))
 
+    from . import transform
     from .load import load_workflow_from_source_code
     from .storage import get_report_by_index
 
     r = get_report_by_index(int(report_id), data)
 
     wf = load_workflow_from_source_code(r.workflow, r.script)
+    cfg = transform.export_config()
+    transform.clear_config()
+    transform.update_config(r.config)
     run_workflow(wf, code, data, plot=plot, freeze=True)
+    transform.clear_config()
+    transform.update_config(cfg)

qulab/executor/transform.py

@@ -31,6 +31,22 @@ def _update_config(updates):
         pickle.dump(parameters, f)
 
 
+def _delete_config(name: str):
+    import pickle
+
+    try:
+        with open('parameters.pkl', 'rb') as f:
+            parameters = pickle.load(f)
+    except:
+        parameters = {}
+
+    if name in parameters:
+        del parameters[name]
+
+    with open('parameters.pkl', 'wb') as f:
+        pickle.dump(parameters, f)
+
+
 def _export_config() -> dict:
     import pickle
 
@@ -43,6 +59,21 @@ def _export_config() -> dict:
     return parameters
 
 
+def _clear_config():
+    import pickle
+
+    try:
+        with open('parameters.pkl', 'rb') as f:
+            parameters = pickle.load(f)
+    except:
+        parameters = {}
+
+    parameters.clear()
+
+    with open('parameters.pkl', 'wb') as f:
+        pickle.dump(parameters, f)
+
+
 def obey_the_oracle(report: Report, data_path):
     global __current_config_id
     update_config(report.oracle)
@@ -67,10 +98,16 @@ def current_config(data_path):
 
 query_config = _query_config
 update_config = _update_config
+delete_config = _delete_config
 export_config = _export_config
+clear_config = _clear_config
 
 
-def set_config_api(query_method, update_method, export_method):
+def set_config_api(query_method,
+                   update_method,
+                   delete_method,
+                   export_method,
+                   clear_method=None):
     """
     Set the query and update methods for the config.
 
@@ -79,13 +116,19 @@ def set_config_api(query_method, update_method, export_method):
             the method should take a key and return the value.
         update_method: The update method.
             the method should take a dict of updates.
+        delete_method: The delete method.
+            the method should take a key and delete it.
         export_method: The export method.
             the method should return a dict of the config.
+        clear_method: The clear method.
+            the method should clear the config.
     """
-    global query_config, update_config, export_config
+    global query_config, update_config, delete_config, export_config, clear_config
 
     query_config = query_method
     update_config = update_method
+    delete_config = delete_method
     export_config = export_method
+    clear_config = clear_method
 
-    return query_config, update_config, export_config
+    return query_config, update_config, delete_config, export_config, clear_config

qulab/version.py

@@ -1 +1 @@
-__version__ = "2.9.6"
+__version__ = "2.9.7"

{qulab-2.9.6.dist-info → qulab-2.9.7.dist-info}/METADATA

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

{qulab-2.9.6.dist-info → qulab-2.9.7.dist-info}/RECORD

@@ -1,20 +1,20 @@
 qulab/__init__.py,sha256=JZ3bn_kTVlnY-P8JXQ5xEdViieFXqfxX8ReLuiiXIpo,321
 qulab/__main__.py,sha256=fjaRSL_uUjNIzBGNgjlGswb9TJ2VD5qnkZHW3hItrD4,68
 qulab/dicttree.py,sha256=tRRMpGZYVOLw0TEByE3_2Ss8FdOmzuGL9e1DWbs8qoY,13684
-qulab/fun.cpython-310-darwin.so,sha256=Sc2xCb7mXN0FJGUY4V15FOXikc7t8yFbtjNYprJJh7Y,126864
+qulab/fun.cpython-310-darwin.so,sha256=VrUlHJ1pFO5oNrnp2dTPjzK0QA-z2vCrZbhevORa26o,126864
 qulab/typing.py,sha256=vg62sGqxuD9CI5677ejlzAmf2fVdAESZCQjAE_xSxPg,69
 qulab/utils.py,sha256=BdLdlfjpe6m6gSeONYmpAKTTqxDaYHNk4exlz8kZxTg,2982
-qulab/version.py,sha256=xbldOz0XM3Jg2OQP4pA5dIy-zqYfu1fHZ9muPBbegtA,21
+qulab/version.py,sha256=YMyATiSQWC8XaFTG2e9YghQWC6WcAk1_57Et_fYwGVs,21
 qulab/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qulab/cli/commands.py,sha256=F1LATmSC9lOAdnrOTMK7DRjETCEcOmMsocovWRyjWTc,597
 qulab/cli/config.py,sha256=Ei7eSYnbwPPlluDnm8YmWONYiI4g7WtvlZGQdr1Z6vo,3688
 qulab/executor/__init__.py,sha256=LosPzOMaljSZY1thy_Fxtbrgq7uubJszMABEB7oM7tU,101
-qulab/executor/cli.py,sha256=OXFmCo2loOCgUYXZpvJz5AnzDag5m_iPetsFM1J11uQ,11291
+qulab/executor/cli.py,sha256=xceK6XNLnNpAVtqvs0_wOAuG0sgS9jg1iTFYmxVUMWY,11795
 qulab/executor/load.py,sha256=qI-3b3xvsjZMCthRO3hlzHA0BZI-tH9Y_L0XLBcyWV8,18657
 qulab/executor/schedule.py,sha256=XRimchHYCgnMAOtCvNLjwMv9IkcTAyBAUCHKQs5RBRw,18745
 qulab/executor/storage.py,sha256=PWQIDYjQaoyLGgAKh0X1tlNQTgDWR8bI-HVie4hSkyA,21075
 qulab/executor/template.py,sha256=_HEtsUQ5_jSujCw8FBDAK1PRTMRCa4iD4DduHIpjo3c,10569
-qulab/executor/transform.py,sha256=ABrnD0l53NI2y2e8ETYO7lmqGtyiE6EAY965kFRqme0,2311
+qulab/executor/transform.py,sha256=rk4CLIKVjGRaFzi5FVSgadUxAKKVLSopEHZCaAzDwDg,3435
 qulab/executor/utils.py,sha256=l_b0y2kMwYKyyXeFtoblPYwKNU-wiFQ9PMo9QlWl9wE,6213
 qulab/monitor/__init__.py,sha256=nTHelnDpxRS_fl_B38TsN0njgq8eVTEz9IAnN3NbDlM,42
 qulab/monitor/__main__.py,sha256=w3yUcqq195LzSnXTkQcuC1RSFRhy4oQ_PEBmucXguME,97
@@ -97,9 +97,9 @@ qulab/visualization/plot_seq.py,sha256=UWTS6p9nfX_7B8ehcYo6UnSTUCjkBsNU9jiOeW2ca
 qulab/visualization/qdat.py,sha256=ZeevBYWkzbww4xZnsjHhw7wRorJCBzbG0iEu-XQB4EA,5735
 qulab/visualization/rot3d.py,sha256=lMrEJlRLwYe6NMBlGkKYpp_V9CTipOAuDy6QW_cQK00,734
 qulab/visualization/widgets.py,sha256=6KkiTyQ8J-ei70LbPQZAK35wjktY47w2IveOa682ftA,3180
-qulab-2.9.6.dist-info/licenses/LICENSE,sha256=PRzIKxZtpQcH7whTG6Egvzl1A0BvnSf30tmR2X2KrpA,1065
-qulab-2.9.6.dist-info/METADATA,sha256=EUA0I36KYGMd3X8yTwXUgOFsCvO4bQx5gZyKnsqfjcU,3720
-qulab-2.9.6.dist-info/WHEEL,sha256=qnIvK8L8kOW9FTrzifubcmNsZckgNfGDNKzGG7FMEdE,114
-qulab-2.9.6.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
-qulab-2.9.6.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
-qulab-2.9.6.dist-info/RECORD,,
+qulab-2.9.7.dist-info/licenses/LICENSE,sha256=PRzIKxZtpQcH7whTG6Egvzl1A0BvnSf30tmR2X2KrpA,1065
+qulab-2.9.7.dist-info/METADATA,sha256=R4asNQDngbtrmIQiaHPnNBGubf76IoqxnODjnKx8GEw,3720
+qulab-2.9.7.dist-info/WHEEL,sha256=qnIvK8L8kOW9FTrzifubcmNsZckgNfGDNKzGG7FMEdE,114
+qulab-2.9.7.dist-info/entry_points.txt,sha256=b0v1GXOwmxY-nCCsPN_rHZZvY9CtTbWqrGj8u1m8yHo,45
+qulab-2.9.7.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
+qulab-2.9.7.dist-info/RECORD,,