track_ids -> segment_ids in charge event gen

DUNE · Dec 21, 2023 · 5ce3e2d · 5ce3e2d
1 parent 9eb95ad
commit 5ce3e2d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/proto_nd_flow/reco/charge/raw_event_generator.py b/src/proto_nd_flow/reco/charge/raw_event_generator.py
@@ -432,8 +432,8 @@ def next(self):
             # write mc data to file
             mc_assn = (np.concatenate(event_mc_assn, axis=0)
                        if len(event_mc_assn) else np.full((0,), -1, dtype=self.mc_assn.dtype))
-            mc_assn_mask = (mc_assn['track_ids'] == -1) | (mc_assn['fraction'] == 0.)
-            event_tracks = ma.array(mc_assn['track_ids'], mask=mc_assn_mask)
+            mc_assn_mask = (mc_assn['segment_ids'] == -1) | (mc_assn['fraction'] == 0.)
+            event_tracks = ma.array(mc_assn['segment_ids'], mask=mc_assn_mask)
             event_packet_fraction = ma.array(mc_assn['fraction'], mask=mc_assn_mask)
 
             # set up packet references