reboost 0.10.0a1__tar.gz → 0.10.0a2__tar.gz

{reboost-0.10.0a1/src/reboost.egg-info → reboost-0.10.0a2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reboost
-Version: 0.10.0a1
+Version: 0.10.0a2
 Summary: New LEGEND Monte-Carlo simulation post-processing
 Author-email: Manuel Huber <info@manuelhu.de>, Toby Dixon <toby.dixon.23@ucl.ac.uk>, Luigi Pertoldi <gipert@pm.me>
 Maintainer: The LEGEND Collaboration

{reboost-0.10.0a1 → reboost-0.10.0a2}/src/reboost/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.10.0a1'
-__version_tuple__ = version_tuple = (0, 10, 0, 'a1')
+__version__ = version = '0.10.0a2'
+__version_tuple__ = version_tuple = (0, 10, 0, 'a2')
 
-__commit_id__ = commit_id = 'g6793c98a5'
+__commit_id__ = commit_id = 'g8f2cabc6a'

{reboost-0.10.0a1 → reboost-0.10.0a2}/src/reboost/build_hit.py

@@ -452,6 +452,8 @@ def _evaluate_operation(
         time_dict=time_dict,
         name=field,
     )
+    if not isinstance(info, str) and "units" in info:
+        col = units.attach_units(col, info["units"])
     units.move_units_to_flattened_data(col)
 
     core.add_field_with_nesting(hit_table, field, col)

{reboost-0.10.0a1 → reboost-0.10.0a2}/src/reboost/core.py

@@ -538,8 +538,5 @@ def remove_columns(tab: Table, outputs: list) -> Table:
 
 def merge(hit_table: Table, output_table: ak.Array | None):
     """Merge the table with the array."""
-    return (
-        hit_table.view_as("ak")
-        if output_table is None
-        else ak.concatenate((output_table, hit_table.view_as("ak")))
-    )
+    hit_table = hit_table.view_as("ak", with_units=True)
+    return hit_table if output_table is None else ak.concatenate((output_table, hit_table))

{reboost-0.10.0a1 → reboost-0.10.0a2/src/reboost.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reboost
-Version: 0.10.0a1
+Version: 0.10.0a2
 Summary: New LEGEND Monte-Carlo simulation post-processing
 Author-email: Manuel Huber <info@manuelhu.de>, Toby Dixon <toby.dixon.23@ucl.ac.uk>, Luigi Pertoldi <gipert@pm.me>
 Maintainer: The LEGEND Collaboration

{reboost-0.10.0a1 → reboost-0.10.0a2}/tests/hit/configs/basic.yaml

@@ -6,11 +6,16 @@ processing_groups:
 
     outputs:
       - t0
+      - t0_u
       - evtid
       - energy
+      - xloc
 
     operations:
       t0: ak.fill_none(ak.firsts(HITS.time, axis=-1), np.nan)
+      t0_u:
+        expression: ak.fill_none(ak.firsts(HITS.time, axis=-1), np.nan)
+        units: ns
       energy: ak.sum(HITS.edep, axis=-1)
 
 forward:

{reboost-0.10.0a1 → reboost-0.10.0a2}/tests/hit/test_build_hit.py

@@ -194,15 +194,20 @@ def test_basic(test_gen_lh5, tmptestdir):
             == b"Zstandard compression: http://www.zstd.net"
         )
 
-    hits = lh5.read("hit/det1", outfile).view_as("ak")
+    hits = lh5.read("hit/det1", outfile).view_as("ak", with_units=True)
 
     assert ak.all(hits.energy == [300, 330])
     assert ak.all(hits.t0 == [0, 0.1])
+
     assert hits.evtid[0] == 0
     assert hits.evtid[1] == 1
 
     assert len(hits) == 2
 
+    assert ak.parameters(hits.t0) == {}
+    assert ak.parameters(hits.t0_u)["units"] == "ns"
+    assert ak.parameters(hits.xloc)["units"] == "m"
+
     # test in memory
 
     hits, time_dict = reboost.build_hit(
@@ -218,6 +223,10 @@ def test_basic(test_gen_lh5, tmptestdir):
     assert ak.all(hits["det1"].evtid[0] == [0, 0])
     assert ak.all(hits["det1"].evtid[1] == [1, 1, 1])
 
+    assert ak.parameters(hits["det1"].t0) == {}
+    assert ak.parameters(hits["det1"].t0_u)["units"] == "ns"
+    assert ak.parameters(hits["det1"].xloc)["units"] == "m"
+
     assert set(time_dict.keys()) == {"global_objects", "geds"}
     assert set(time_dict["geds"].keys()) == {
         "detector_objects",
@@ -226,7 +235,7 @@ def test_basic(test_gen_lh5, tmptestdir):
         "expressions",
     }
     assert set(time_dict["geds"]["read"].keys()) == {"stp"}
-    assert set(time_dict["geds"]["expressions"].keys()) == {"t0", "energy"}
+    assert set(time_dict["geds"]["expressions"].keys()) == {"t0", "t0_u", "energy"}
 
 
 def test_file_merging(test_gen_lh5, tmptestdir):