From e6094d30287c0f1993d11b3e8b228d45b1fe2040 Mon Sep 17 00:00:00 2001 From: Rasmus Oersoe Date: Mon, 20 Mar 2023 14:16:08 +0100 Subject: [PATCH] push fix --- src/graphnet/data/extractors/i3featureextractor.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/src/graphnet/data/extractors/i3featureextractor.py b/src/graphnet/data/extractors/i3featureextractor.py index 53d3a116e..3926b38ec 100644 --- a/src/graphnet/data/extractors/i3featureextractor.py +++ b/src/graphnet/data/extractors/i3featureextractor.py @@ -148,9 +148,15 @@ def __call__(self, frame: "icetray.I3Frame") -> Dict[str, List[Any]]: output["is_errata_dom"].append(is_errata_dom) output["event_time"].append(event_time) # Pulse flags - output["hlc"].append((pulse.flags >> 0) & 0x1) # bit 0 - output["awtd"].append((pulse.flags >> 1) & 0x1) # bit 1 - output["fadc"].append((pulse.flags >> 2) & 0x1) # bit 2 + flags = getattr(pulse, "flags", padding_value) + if flags == padding_value: + output["hlc"].append(padding_value) # bit 0 + output["awtd"].append(padding_value) # bit 1 + output["fadc"].append(padding_value) # bit 2 + else: + output["hlc"].append((pulse.flags >> 0) & 0x1) # bit 0 + output["awtd"].append((pulse.flags >> 1) & 0x1) # bit 1 + output["fadc"].append((pulse.flags >> 2) & 0x1) # bit 2 return output