numbox 0.2.2__py3-none-any.whl → 0.2.3__py3-none-any.whl

numbox/__init__.py

@@ -1 +1 @@
-__version__ = '0.2.2'
+__version__ = '0.2.3'

numbox/core/work/work.py

@@ -3,11 +3,12 @@ from io import StringIO
 from numba import njit
 from numba.core.errors import NumbaError
 from numba.core.types import (
-    DictType, FunctionType, Integer, Literal, NoneType, Tuple, unicode_type, UnicodeType
+    boolean, DictType, FunctionType, Literal, NoneType, Tuple, unicode_type, UnicodeType
 )
 from numba.core.typing.context import Context
 from numba.experimental.structref import define_boxing, new
 from numba.extending import intrinsic, overload, overload_method
+from numba.typed.typeddict import Dict
 from numba.typed.typedlist import List
 
 from numbox.core.any.erased_type import ErasedType
@@ -291,12 +292,14 @@ def _make_combine_code(num_sources):
     for source_ind_ in range(num_sources):
         code_txt.write(_make_source_getter(source_ind_))
     code_txt.write("""
-def _combine_(work_, data_, ct_=0):
-    if ct_ == len(data_):
-        return ct_
+def _combine_(work_, data_, harvested_=None):
+    if harvested_ is None:
+        harvested_ = Dict.empty(key_type=unicode_type, value_type=boolean)
+    if len(harvested_) == len(data_):
+        return
     work_name = work_.name
     if work_name in data_:
-        ct_ += 1
+        harvested_[work_name] = True
         data_[work_name].reset(work_.data)""")
     if num_sources > 0:
         code_txt.write("""
@@ -304,10 +307,10 @@ def _combine_(work_, data_, ct_=0):
         for source_ind_ in range(num_sources):
             code_txt.write(f"""
     source_{source_ind_} = _get_source_{source_ind_}(sources)
-    ct_ = source_{source_ind_}.combine(data_, ct_)
+    source_{source_ind_}.combine(data_, harvested_)
 """)
     code_txt.write("""
-    return ct_
+    return
 """)
     return code_txt.getvalue()
 
@@ -316,7 +319,7 @@ _combine_registry = {}
 
 
 @overload_method(WorkTypeClass, "combine", strict=False, jit_options=default_jit_options)
-def ol_combine(work_ty, data_ty: DictType, ct_ty=Integer):
+def ol_combine(work_ty, data_ty: DictType, harvested_ty=NoneType):
     """ Harvest nodes data from the graph with the root node `work`.
      `data` is provided as dictionary mapping node name to `Any` type
      containing erased payload `p` to be reset to `data`. """
@@ -325,7 +328,7 @@ def ol_combine(work_ty, data_ty: DictType, ct_ty=Integer):
     _combine = _combine_registry.get(num_sources, None)
     if _combine is None:
         code_txt = _make_combine_code(num_sources)
-        ns = getmodule(_file_anchor).__dict__
+        ns = {**getmodule(_file_anchor).__dict__, **{"boolean": boolean, "Dict": Dict}}
         code = compile(code_txt, getfile(_file_anchor), mode="exec")
         exec(code, ns)
         _combine = ns["_combine_"]

{numbox-0.2.2.dist-info → numbox-0.2.3.dist-info}/METADATA

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

{numbox-0.2.2.dist-info → numbox-0.2.3.dist-info}/RECORD

@@ -1,4 +1,4 @@
-numbox/__init__.py,sha256=QQ0yjZQnvXM7JeLLZEZWX6NmVUPu1lSROWOXF49_gug,22
+numbox/__init__.py,sha256=XtWUl6HPylv5jZLd2KkgtPptuzuda93kC2REmOrF-Cs,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
@@ -22,7 +22,7 @@ numbox/core/work/lowlevel_work_utils.py,sha256=TgRRcNfks0oaOXGXXr3ptafd_Xv_lpmH8
 numbox/core/work/node.py,sha256=CMolyoRQjG2A-pTQqZQ0kxKOYTKipWRC0mu8RWHuTUI,5096
 numbox/core/work/node_base.py,sha256=uI7asM2itQcHuOByXyJtqvrd4ovW6EXDRdHYp3JVHQ0,998
 numbox/core/work/print_tree.py,sha256=y2u7xmbHvpcA57y8PrGSqOunLNCqhgNXdVtXHqvy1M0,2340
-numbox/core/work/work.py,sha256=OZ9rRbaOAzQhvw2akORSlqPllwRe1BUhkMFWZOJSmo0,14369
+numbox/core/work/work.py,sha256=oC71F2y-NkIF9Ox63trlv061nYBxla6R-F2Nj-7NXI4,14595
 numbox/core/work/work_utils.py,sha256=3q_nnBdzuxWWcdFpbRL2H0T9ZNkUgx1J1uhiZkX3YG4,1039
 numbox/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 numbox/utils/highlevel.py,sha256=gXYzLsFPKHQfGqaz-Z4DWcCQddv0dS6SKwIsM-_xjYg,1487
@@ -30,8 +30,8 @@ numbox/utils/lowlevel.py,sha256=ACpf8_HyOIsobPlZ31bapkEyuCsV5dojW3AFrcKykrw,1071
 numbox/utils/meminfo.py,sha256=ykFi8Vt0WcHI3ztgMwvpn6NqaflDSQGL8tjI01jrzm0,1759
 numbox/utils/timer.py,sha256=KkAkWOHQ72WtPjyiAzt_tF1q0DcOnCDkITTb85DvkUM,553
 numbox/utils/void_type.py,sha256=IkZsjNeAIShYJtvWbvERdHnl_mbF1rCRWiM3gp6II8U,404
-numbox-0.2.2.dist-info/LICENSE,sha256=YYgNvjH_p6-1NsdrIqGJnr1GUbZzA_8DxsP6vVfM6nY,1446
-numbox-0.2.2.dist-info/METADATA,sha256=mbgQDgqnezAK4URoAhC4C4BHwCy5ThKSDkdTkkD6WVQ,2792
-numbox-0.2.2.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-numbox-0.2.2.dist-info/top_level.txt,sha256=A67jOkfqidCSYYm6ifjN_WZyIiR1B27fjxv6nNbPvjc,7
-numbox-0.2.2.dist-info/RECORD,,
+numbox-0.2.3.dist-info/LICENSE,sha256=YYgNvjH_p6-1NsdrIqGJnr1GUbZzA_8DxsP6vVfM6nY,1446
+numbox-0.2.3.dist-info/METADATA,sha256=e7MF4nI2ntAqX8CGKWS1FvUiEswd2rtRW-rysf4WWJc,2792
+numbox-0.2.3.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+numbox-0.2.3.dist-info/top_level.txt,sha256=A67jOkfqidCSYYm6ifjN_WZyIiR1B27fjxv6nNbPvjc,7
+numbox-0.2.3.dist-info/RECORD,,