numbox 0.2.8__py3-none-any.whl → 0.2.9__py3-none-any.whl

numbox/__init__.py

@@ -1 +1 @@
-__version__ = '0.2.8'
+__version__ = '0.2.9'

numbox/core/work/builder.py

@@ -22,16 +22,18 @@ _specs_registry = dict()
 class _End(NamedTuple):
     name: str
     init_value: Any
+    registry: dict = None
     ty: Optional[type | Type] = None
 
 
 def _new(cls, super_proxy, *args, **kwargs):
     name = kwargs.get("name")
     assert name, "`name` key-word argument has not been provided"
-    if name in _specs_registry:
+    registry = kwargs.get("registry", _specs_registry)
+    if name in registry:
         raise ValueError(f"Node '{name}' has already been defined on this graph. Pick a different name.")
     spec_ = super_proxy.__new__(cls, *args, **kwargs)
-    _specs_registry[name] = spec_
+    registry[name] = spec_
     return spec_
 
 
@@ -47,6 +49,7 @@ class _Derived(NamedTuple):
     init_value: Any
     derive: Callable
     sources: Sequence[Union['Derived', End]]
+    registry: dict = None
     ty: Optional[type | Type] = None
 
 
@@ -118,29 +121,35 @@ def code_block_hash(code_txt: str):
     return sha256(code_txt.encode("utf-8")).hexdigest()
 
 
-def _infer_end_and_derived_nodes(spec: SpecTy, all_inputs_: Dict[str, Type], all_derived_: Dict[str, Type]):
+def _infer_end_and_derived_nodes(spec: SpecTy, all_inputs_: Dict[str, Type], all_derived_: Dict[str, Type], registry):
     if spec.name in all_inputs_ or spec.name in all_derived_:
         return
     if isinstance(spec, End):
         all_inputs_[spec.name] = get_ty(spec)
         return
     for source in spec.sources:
-        _infer_end_and_derived_nodes(source, all_inputs_, all_derived_)
+        _infer_end_and_derived_nodes(source, all_inputs_, all_derived_, registry)
     all_derived_[spec.name] = get_ty(spec)
 
 
-def infer_end_and_derived_nodes(access_nodes: SpecTy | Sequence[SpecTy]):
+def infer_end_and_derived_nodes(access_nodes: SpecTy | Sequence[SpecTy], registry):
     all_inputs_ = dict()
     all_derived_ = dict()
     for access_node in access_nodes:
-        _infer_end_and_derived_nodes(access_node, all_inputs_, all_derived_)
-    all_inputs_lst = [_specs_registry[name] for name in all_inputs_.keys()]
-    all_derived_lst = [_specs_registry[name] for name in all_derived_.keys()]
+        _infer_end_and_derived_nodes(access_node, all_inputs_, all_derived_, registry)
+    all_inputs_lst = [registry[name] for name in all_inputs_.keys()]
+    all_derived_lst = [registry[name] for name in all_derived_.keys()]
     return all_inputs_lst, all_derived_lst
 
 
-def make_graph(*access_nodes: SpecTy | Sequence[SpecTy], jit_options: Optional[dict] = None):
-    all_inputs_, all_derived_ = infer_end_and_derived_nodes(access_nodes)
+def make_graph(
+    *access_nodes: SpecTy | Sequence[SpecTy],
+    registry: Optional[dict] = None,
+    jit_options: Optional[dict] = None
+):
+    if registry is None:
+        registry = _specs_registry
+    all_inputs_, all_derived_ = infer_end_and_derived_nodes(access_nodes, registry)
     if jit_options is None:
         jit_options = {}
     jit_options = {**default_jit_options, **jit_options}

{numbox-0.2.8.dist-info → numbox-0.2.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: numbox
-Version: 0.2.8
+Version: 0.2.9
 Author: Mikhail Goykhman
 License: MIT License (with Citation Clause)
         

{numbox-0.2.8.dist-info → numbox-0.2.9.dist-info}/RECORD

@@ -1,4 +1,4 @@
-numbox/__init__.py,sha256=QS1piWKz3Mys2vOiR2kTlrqfdbiwJYEfhOlgbh73Sg8,22
+numbox/__init__.py,sha256=bd8hlxKMc0Fh_2w5GWfqBtuwa8uTepTMjT8w7F-EKvY,22
 numbox/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 numbox/core/configurations.py,sha256=0bCmxXL-QMwtvyIDhpXLeT-1KJMf_QpH0wLuEvYLGxQ,68
 numbox/core/any/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -15,7 +15,7 @@ numbox/core/bindings/utils.py,sha256=aRtN8oUYBk9vgoUGaUJosGx0Za-vvCNwwbZg_g_-LRs
 numbox/core/proxy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 numbox/core/proxy/proxy.py,sha256=Wt7yzswDmeQXt0yjcTcnLi2coneowSHWXy_IFpZZJMU,3612
 numbox/core/work/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-numbox/core/work/builder.py,sha256=U7hxopSwOc1ke1Hqup6xCliO7bqjq91zVX1SixYPxoM,6072
+numbox/core/work/builder.py,sha256=jH9lIxpJcU-eJ8SMQe7TzV8xci7FQgK8ZOhFma0n5DI,6308
 numbox/core/work/combine_utils.py,sha256=qTVGke_ydzaTQ7o29DFjZWZzKjRNKb0L3yJMaR3TLII,2430
 numbox/core/work/explain.py,sha256=ESwvsTgfe0w7UnM13yyVpVDtfJyAK2A1sNdF3RNb-jU,1200
 numbox/core/work/loader_utils.py,sha256=g83mDWidZJ8oLWP3I3rK8aGISYOO2S-w6HDgtosCyck,1572
@@ -32,8 +32,8 @@ numbox/utils/meminfo.py,sha256=ykFi8Vt0WcHI3ztgMwvpn6NqaflDSQGL8tjI01jrzm0,1759
 numbox/utils/standard.py,sha256=2fPrMlSXe2TG3CIfjJOT8LQkHEH86oOOj1AvwQkYCfA,450
 numbox/utils/timer.py,sha256=5_d690Fb3L2axJBRxtoB0qe23exBosNR4qu6cno4QfY,641
 numbox/utils/void_type.py,sha256=IkZsjNeAIShYJtvWbvERdHnl_mbF1rCRWiM3gp6II8U,404
-numbox-0.2.8.dist-info/LICENSE,sha256=YYgNvjH_p6-1NsdrIqGJnr1GUbZzA_8DxsP6vVfM6nY,1446
-numbox-0.2.8.dist-info/METADATA,sha256=krcMK7v4WNgTrcuriQAWMASeaRod9KybPc57tRUE7w0,2935
-numbox-0.2.8.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-numbox-0.2.8.dist-info/top_level.txt,sha256=A67jOkfqidCSYYm6ifjN_WZyIiR1B27fjxv6nNbPvjc,7
-numbox-0.2.8.dist-info/RECORD,,
+numbox-0.2.9.dist-info/LICENSE,sha256=YYgNvjH_p6-1NsdrIqGJnr1GUbZzA_8DxsP6vVfM6nY,1446
+numbox-0.2.9.dist-info/METADATA,sha256=s-0iSTOHKbHEyDL5u8DgOPcbgHSs1897HV9T3963x9E,2935
+numbox-0.2.9.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+numbox-0.2.9.dist-info/top_level.txt,sha256=A67jOkfqidCSYYm6ifjN_WZyIiR1B27fjxv6nNbPvjc,7
+numbox-0.2.9.dist-info/RECORD,,