QuLab 2.2.8__cp310-cp310-macosx_10_9_universal2.whl → 2.3.1__cp310-cp310-macosx_10_9_universal2.whl

{QuLab-2.2.8.dist-info → QuLab-2.3.1.dist-info}/METADATA

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

{QuLab-2.2.8.dist-info → QuLab-2.3.1.dist-info}/RECORD

@@ -1,7 +1,7 @@
 qulab/__init__.py,sha256=P-Mx2p4TVmL91SoxoeXcj8Qm0x4xUf5Q_FLk0Yc_gIQ,138
 qulab/__main__.py,sha256=ZC1NKaoxKyy60DaCfB8vYnB1z3RXQ2j8E1sRZ4A8sXE,428
-qulab/fun.cpython-310-darwin.so,sha256=FGbpFs23OHVvbhO8uG5EqVKCp7tBZrlCXe7dRshO47k,159632
-qulab/version.py,sha256=orWdhtLf5L0uyMwYMv-baYndrBUtQQg-ackBu8Y5UJo,21
+qulab/fun.cpython-310-darwin.so,sha256=VlJHvNsGpb8r_ZqQcd1fX7jR_Q5KnbIL-uSOP214K-0,159632
+qulab/version.py,sha256=6STsl4r4R0vt1cK4ruxjle9q351XfLSSyARO42RdqOk,21
 qulab/monitor/__init__.py,sha256=nTHelnDpxRS_fl_B38TsN0njgq8eVTEz9IAnN3NbDlM,42
 qulab/monitor/__main__.py,sha256=w3yUcqq195LzSnXTkQcuC1RSFRhy4oQ_PEBmucXguME,97
 qulab/monitor/config.py,sha256=fQ5JcsMApKc1UwANEnIvbDQZl8uYW0tle92SaYtX9lI,744
@@ -14,14 +14,14 @@ qulab/monitor/qt_compat.py,sha256=OK71_JSO_iyXjRIKHANmaK4Lx4bILUzmXI-mwKg3QeI,78
 qulab/monitor/toolbar.py,sha256=WEag6cxAtEsOLL14XvM7pSE56EA3MO188_JuprNjdBs,7948
 qulab/scan/__init__.py,sha256=ZX4WsvqYxvJeHLgGSrtJoAnVU94gxY7EHKMxYooMERg,130
 qulab/scan/curd.py,sha256=thq_qfi3qng3Zx-1uhNG64IQhGCuum_LR4MOKnS8cDI,6896
-qulab/scan/expression.py,sha256=JsJ00qK05QVDAgz-husRuov7hKg8nW75rsTwLG8RH7Q,15843
+qulab/scan/expression.py,sha256=s2_sWFm4gKgJ7pGLmuugCU8XifAm9Mc3AT3bdoyFUtM,19459
 qulab/scan/models.py,sha256=5Jpo25WGMWs0GtLzYLsWO61G3-FFYx5BHhBr2b6rOTE,17681
-qulab/scan/optimize.py,sha256=vErjRTCtn2MwMF5Xyhs1P4gHF2IFHv_EqxsUvH_4y7k,2287
+qulab/scan/optimize.py,sha256=jYceGazEabuwYqmZE8P6Nnq_KsLyUP0p-p88VbbJLhM,2559
 qulab/scan/query.py,sha256=-5uHMhXSyGovK1oy_uUbGVEbRFzaMBkP78ZMNfI3jD0,11809
-qulab/scan/record.py,sha256=-FL9JMgfhvkT0q7dhJJXbjXOrOkTIk2J9oPmxP6psuA,21070
-qulab/scan/scan.py,sha256=KRCoq0ezqwIQlOifUAk9hUR-G-ryywUwEwP__rLMdPo,36499
+qulab/scan/record.py,sha256=sICMLAWf3IgvJ3JlS0cKQA4mSpLZins7K5qFoAnCSYI,21194
+qulab/scan/scan.py,sha256=GXjx5gDT2L7S2H_opdFl9fEEJNsHi2RlwJpcbAY4aKg,37837
 qulab/scan/server.py,sha256=ZT-J447xuLSpmbTNm5dNIz8_CXKOow66_bjkFNGh_vI,14274
-qulab/scan/space.py,sha256=7Won0i4-5wuFeIACFx2Lq3BAqc_3S4S_nuVaSDTSeZs,5293
+qulab/scan/space.py,sha256=X5anTvw7X0XGN83y0o0C-zhvh6bDEU-kshGgo004pNE,6076
 qulab/scan/utils.py,sha256=Pg_tCf3SUKTiPSBqb6Enkgx4bAyQJAkDGe9uYys1xVU,3613
 qulab/storage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qulab/storage/__main__.py,sha256=3emxxRry8BB0m8hUZvJ_oBqkPy7ksV7flHB_KEDXZuI,1692
@@ -78,9 +78,9 @@ qulab/visualization/plot_layout.py,sha256=clNw9QjE_kVNpIIx2Ob4YhAz2fucPGMuzkoIrO
 qulab/visualization/plot_seq.py,sha256=lphYF4VhkEdc_wWr1kFBwrx2yujkyFPFaJ3pjr61awI,2693
 qulab/visualization/qdat.py,sha256=ZeevBYWkzbww4xZnsjHhw7wRorJCBzbG0iEu-XQB4EA,5735
 qulab/visualization/widgets.py,sha256=6KkiTyQ8J-ei70LbPQZAK35wjktY47w2IveOa682ftA,3180
-QuLab-2.2.8.dist-info/LICENSE,sha256=PRzIKxZtpQcH7whTG6Egvzl1A0BvnSf30tmR2X2KrpA,1065
-QuLab-2.2.8.dist-info/METADATA,sha256=0YYD14QsQsGNgqZ-Wo26Y6N3Fd47CbE501cTGU0vah4,3510
-QuLab-2.2.8.dist-info/WHEEL,sha256=r7U64H7df5k5VoE41bE2otJ6YmrMps4wUd2_S2hHvHQ,115
-QuLab-2.2.8.dist-info/entry_points.txt,sha256=ohBzutEnQimP_BZWiuXdSliu4QAYSHHcN0PZD8c7ZCY,46
-QuLab-2.2.8.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
-QuLab-2.2.8.dist-info/RECORD,,
+QuLab-2.3.1.dist-info/LICENSE,sha256=PRzIKxZtpQcH7whTG6Egvzl1A0BvnSf30tmR2X2KrpA,1065
+QuLab-2.3.1.dist-info/METADATA,sha256=86_MtpxRFwbah3FMBNV24Gsr_p05NkZo6Sx6904TtUw,3510
+QuLab-2.3.1.dist-info/WHEEL,sha256=ClVOi9g10rI0HLG8M8rsO0ZIZ1smbNYM10V9YDSXCJY,114
+QuLab-2.3.1.dist-info/entry_points.txt,sha256=ohBzutEnQimP_BZWiuXdSliu4QAYSHHcN0PZD8c7ZCY,46
+QuLab-2.3.1.dist-info/top_level.txt,sha256=3T886LbAsbvjonu_TDdmgxKYUn939BVTRPxPl9r4cEg,6
+QuLab-2.3.1.dist-info/RECORD,,

{QuLab-2.2.8.dist-info → QuLab-2.3.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.43.0)
+Generator: setuptools (70.1.0)
 Root-Is-Purelib: false
 Tag: cp310-cp310-macosx_10_9_universal2
 

qulab/scan/expression.py

@@ -210,6 +210,16 @@ class Expression():
             return -self
         return BinaryExpression(other, self, operator.mul)
 
+    def __matmul__(self, other):
+        if isinstance(other, Expression):
+            other = other.eval(_default_env)
+        return BinaryExpression(self, other, operator.matmul)
+
+    def __rmatmul__(self, other):
+        if isinstance(other, Expression):
+            other = other.eval(_default_env)
+        return BinaryExpression(other, self, operator.matmul)
+
     def __truediv__(self, other):
         if isinstance(other, Expression):
             other = other.eval(_default_env)
@@ -226,6 +236,34 @@ class Expression():
             return 0
         return BinaryExpression(other, self, operator.truediv)
 
+    def __floordiv__(self, other):
+        if isinstance(other, Expression):
+            other = other.eval(_default_env)
+        if isinstance(other, ConstType) and other == 1:
+            return self
+        if isinstance(other, ConstType) and other == -1:
+            return -self
+        return BinaryExpression(self, other, operator.floordiv)
+
+    def __rfloordiv__(self, other):
+        if isinstance(other, Expression):
+            other = other.eval(_default_env)
+        if isinstance(other, ConstType) and other == 0:
+            return 0
+        return BinaryExpression(other, self, operator.floordiv)
+
+    def __mod__(self, other):
+        if isinstance(other, Expression):
+            other = other.eval(_default_env)
+        if isinstance(other, ConstType) and other == 1:
+            return 0
+        return BinaryExpression(self, other, operator.mod)
+
+    def __rmod__(self, other):
+        if isinstance(other, Expression):
+            other = other.eval(_default_env)
+        return BinaryExpression(other, self, operator.mod)
+
     def __pow__(self, other):
         if isinstance(other, Expression):
             other = other.eval(_default_env)
@@ -248,6 +286,21 @@ class Expression():
     def __pos__(self):
         return UnaryExpression(self, operator.pos)
 
+    def __abs__(self):
+        return UnaryExpression(self, operator.abs)
+
+    def __not__(self):
+        return UnaryExpression(self, operator.not_)
+
+    def __inv__(self):
+        return UnaryExpression(self, operator.inv)
+
+    def __invert__(self):
+        return UnaryExpression(self, operator.invert)
+
+    def __index__(self):
+        return UnaryExpression(self, operator.index)
+
     def __eq__(self, other):
         if isinstance(other, Expression):
             other = other.eval(_default_env)
@@ -278,6 +331,56 @@ class Expression():
             other = other.eval(_default_env)
         return BinaryExpression(self, other, operator.ge)
 
+    def __and__(self, other):
+        if isinstance(other, Expression):
+            other = other.eval(_default_env)
+        return BinaryExpression(self, other, operator.and_)
+
+    def __rand__(self, other):
+        if isinstance(other, Expression):
+            other
+        return BinaryExpression(other, self, operator.and_)
+
+    def __or__(self, other):
+        if isinstance(other, Expression):
+            other = other.eval(_default_env)
+        return BinaryExpression(self, other, operator.or_)
+
+    def __ror__(self, other):
+        if isinstance(other, Expression):
+            other = other.eval(_default_env)
+        return BinaryExpression(other, self, operator.or_)
+
+    def __lshift__(self, other):
+        if isinstance(other, Expression):
+            other = other.eval(_default_env)
+        return BinaryExpression(self, other, operator.lshift)
+
+    def __rlshift__(self, other):
+        if isinstance(other, Expression):
+            other = other.eval(_default_env)
+        return BinaryExpression(other, self, operator.lshift)
+
+    def __rshift__(self, other):
+        if isinstance(other, Expression):
+            other = other.eval(_default_env)
+        return BinaryExpression(self, other, operator.rshift)
+
+    def __rrshift__(self, other):
+        if isinstance(other, Expression):
+            other = other.eval(_default_env)
+        return BinaryExpression(other, self, operator.rshift)
+
+    def __xor__(self, other):
+        if isinstance(other, Expression):
+            other = other.eval(_default_env)
+        return BinaryExpression(self, other, operator.xor)
+
+    def __rxor__(self, other):
+        if isinstance(other, Expression):
+            other = other.eval(_default_env)
+        return BinaryExpression(other, self, operator.xor)
+
     def __getitem__(self, other):
         if isinstance(other, Expression):
             other = other.eval(_default_env)
@@ -301,6 +404,9 @@ class Expression():
     def __round__(self, n=None):
         return self
 
+    def __bool__(self):
+        return True
+
     def eval(self, env):
         raise NotImplementedError
 

qulab/scan/optimize.py

@@ -37,6 +37,12 @@ class NgOptimizer():
                            self.config['method'])(self.instrum,
                                                   budget=self.config['budget'])
 
+    def suggest(self, *suggested):
+        suggested = [
+            space.transform(x) for x, space in zip(suggested, self.dimensions)
+        ]
+        self.opt.suggest(*suggested)
+
     def ask(self):
         tmp = self.opt.ask()
         return [
@@ -47,15 +53,16 @@ class NgOptimizer():
     def tell(self, suggested: Sequence, value: Any):
         self._all_x.append(suggested)
         self._all_y.append(value)
-        suggested = [
+        suggested = tuple([
             space.transform(x) for x, space in zip(suggested, self.dimensions)
-        ]
-        self.opt.suggest(*suggested)
-        x = self.opt.ask()
+        ])
+        # self.opt.suggest(*suggested)
+        # x = self.opt.ask()
+        x = self.instrum.spawn_child(new_value=(suggested, {}))
         self.opt.tell(x, value)
 
     def get_result(self, history: bool = False) -> OptimizeResult:
-        recommendation = self.opt.recommend()
+        recommendation = self.opt.provide_recommendation()
         ret = OptimizeResult({
             'x': [
                 space.inverse_transform(x)

qulab/scan/record.py

@@ -589,7 +589,9 @@ class Record():
         return record
 
     def __repr__(self):
-        return f"<Record: id={self.id} app={self.description['app']}, keys={self.keys()}>"
+        if self.is_remote_record() and not self._items:
+            self._items = {key: None for key in self.keys()}
+        return f"<Record: id={self.id} app={self.description['app']}, keys={self._items.keys()}>"
 
     # def _repr_html_(self):
     #     return f"""

qulab/scan/scan.py

@@ -226,10 +226,11 @@ class Scan():
                  max_workers: int = 4,
                  max_promise: int = 100,
                  max_message: int = 1000,
+                 config: dict | None = None,
                  mixin=None):
         self.id = task_uuid()
         self.record = None
-        self.config = None
+        self.config = {} if config is None else copy.deepcopy(config)
         self.description = {
             'app': app,
             'tags': tags,
@@ -250,7 +251,7 @@ class Scan():
             'filters': {},
             'total': {},
             'database': database,
-            'hiden': ['self', r'^__.*', r'.*__$'],
+            'hiden': ['self', 'config', r'^__.*', r'.*__$'],
             'entry': {
                 'system': get_system_info(),
                 'env': {},
@@ -392,7 +393,10 @@ class Scan():
         if name in self.description['consts']:
             return self.description['consts'][name]
         else:
-            return Symbol(name)
+            try:
+                return self._query_config(name)
+            except:
+                return Symbol(name)
 
     def _add_search_space(self, name: str, level: int, space):
         if level not in self.description['loops']:
@@ -402,6 +406,8 @@ class Scan():
     def add_depends(self, name: str, depends: list[str]):
         if isinstance(depends, str):
             depends = [depends]
+        if 'self' in depends:
+            depends.append('config')
         if name not in self.description['dependents']:
             self.description['dependents'][name] = set()
         self.description['dependents'][name].update(depends)
@@ -448,6 +454,9 @@ class Scan():
                 pass
             self.description['consts'][name] = value
 
+        if '.' in name:
+            self.add_depends('config', [name])
+
         if ',' in name:
             for key in name.split(','):
                 if not key.startswith('*'):
@@ -472,6 +481,8 @@ class Scan():
         elif isinstance(space, OptimizeSpace):
             space.name = name
             space.optimizer.dimensions[name] = space.space
+            if space.suggestions:
+                space.optimizer.suggestions[name] = space.suggestions
             self._add_search_space(name, space.optimizer.level, space)
             self.add_depends(space.optimizer.name, [name])
         else:
@@ -486,6 +497,8 @@ class Scan():
                 self.add_depends(name, space.symbols())
         if setter:
             self.description['setters'][name] = setter
+        if '.' in name:
+            self.add_depends('config', [name])
 
     def trace(self,
               name: str,
@@ -535,6 +548,28 @@ class Scan():
             self.description['getters'][name] = getter
         return opt
 
+    def _synchronize_config(self):
+        for key, value in self.variables.items():
+            if '.' in key:
+                d = self.config
+                ks = key.split('.')
+                if not ks:
+                    continue
+                for k in ks[:-1]:
+                    if k in d:
+                        d = d[k]
+                    else:
+                        d[k] = {}
+                        d = d[k]
+                d[ks[-1]] = value
+        return self.config
+
+    def _query_config(self, key):
+        d = self.config
+        for k in key.split('.'):
+            d = d[k]
+        return d
+
     async def _update_progress(self):
         while True:
             task = await self._prm_queue.get()
@@ -549,17 +584,6 @@ class Scan():
 
     async def run(self):
         assymbly(self.description)
-        if self.config:
-            self.description['config'] = await create_config(
-                self.config, self.description['database'], self._sock)
-        if current_notebook() is None:
-            await create_notebook('untitle', self.description['database'],
-                                  self._sock)
-        cell_id = await save_input_cells(current_notebook(),
-                                         self.description['entry']['scripts'],
-                                         self.description['database'],
-                                         self._sock)
-        self.description['entry']['scripts'] = cell_id
         if isinstance(
                 self.description['database'],
                 str) and self.description['database'].startswith("tcp://"):
@@ -567,15 +591,28 @@ class Scan():
                                          connect=self.description['database'],
                                          socket=self._sock) as socket:
                 self._sock = socket
+                if self.config:
+                    self.description['config'] = await create_config(
+                        self.config, self.description['database'], self._sock)
+                if current_notebook() is None:
+                    await create_notebook('untitle',
+                                          self.description['database'],
+                                          self._sock)
+                cell_id = await save_input_cells(
+                    current_notebook(), self.description['entry']['scripts'],
+                    self.description['database'], self._sock)
+                self.description['entry']['scripts'] = cell_id
                 await self._run()
         else:
+            if self.config:
+                self.description['config'] = copy.deepcopy(self.config)
             await self._run()
 
     async def _run(self):
         send_msg_task = asyncio.create_task(self._send_msg())
         update_progress_task = asyncio.create_task(self._update_progress())
 
-        self._variables = {'self': self}
+        self._variables = {'self': self, 'config': self.config}
 
         consts = {}
         for k, v in self.description['consts'].items():
@@ -671,8 +708,10 @@ class Scan():
                 self.variables,
                 self.description['loops'].get(self.current_level, []),
                 self.description['order'].get(self.current_level, []),
-                self.description['functions'], self.description['optimizers'],
-                self.description['setters'], self.description['getters']):
+                self.description['functions']
+                | {'config': self._synchronize_config},
+                self.description['optimizers'], self.description['setters'],
+                self.description['getters']):
             self._current_level += 1
             if await self._filter(variables, self.current_level - 1):
                 yield variables

qulab/scan/space.py

@@ -1,3 +1,4 @@
+import itertools
 from typing import Type
 
 import numpy as np
@@ -103,10 +104,14 @@ def geomspace(start, stop, num=50, endpoint=True):
 
 class OptimizeSpace():
 
-    def __init__(self, optimizer: 'Optimizer', space):
+    def __init__(self, optimizer: 'Optimizer', space, suggestion=None):
         self.optimizer = optimizer
         self.space = space
         self.name = None
+        if suggestion is not None and not isinstance(
+                suggestion, (list, tuple, np.ndarray)):
+            suggestion = [suggestion]
+        self.suggestion = suggestion
 
     def __len__(self):
         return self.optimizer.maxiter
@@ -130,17 +135,32 @@ class Optimizer():
         self.level = level
         self.kwds = kwds
         self.minimize = minimize
+        self.suggestion = {}
 
     def create(self):
-        return self.method(list(self.dimensions.values()), **self.kwds)
+        opt = self.method(list(self.dimensions.values()), **self.kwds)
+
+        def rvs(space):
+            while True:
+                yield space.rvs()[0]
+
+        if self.suggestion:
+            for suggestion in zip(*[
+                    self.suggestion.get(key, rvs(space))
+                    for key, space in self.dimensions.items()
+            ]):
+                opt.suggest(*suggestion)
+        return opt
 
     def Categorical(self,
                     categories,
                     prior=None,
                     transform=None,
-                    name=None) -> OptimizeSpace:
+                    name=None,
+                    suggestion=None) -> OptimizeSpace:
         return OptimizeSpace(self,
-                             Categorical(categories, prior, transform, name))
+                             Categorical(categories, prior, transform, name),
+                             suggestion)
 
     def Integer(self,
                 low,
@@ -149,9 +169,11 @@ class Optimizer():
                 base=10,
                 transform="normalize",
                 name=None,
-                dtype=np.int64) -> OptimizeSpace:
+                dtype=np.int64,
+                suggestion=None) -> OptimizeSpace:
         return OptimizeSpace(
-            self, Integer(low, high, prior, base, transform, name, dtype))
+            self, Integer(low, high, prior, base, transform, name, dtype),
+            suggestion)
 
     def Real(self,
              low,
@@ -160,9 +182,11 @@ class Optimizer():
              base=10,
              transform="normalize",
              name=None,
-             dtype=float) -> OptimizeSpace:
+             dtype=float,
+             suggestion=None) -> OptimizeSpace:
         return OptimizeSpace(
-            self, Real(low, high, prior, base, transform, name, dtype))
+            self, Real(low, high, prior, base, transform, name, dtype),
+            suggestion)
 
     def __getstate__(self) -> dict:
         state = self.__dict__.copy()

qulab/version.py

@@ -1 +1 @@
-__version__ = "2.2.8"
+__version__ = "2.3.1"