From f30eb56c6a79fd115d64192a8081d661b94cf75b Mon Sep 17 00:00:00 2001 From: "Marcel R." Date: Mon, 13 Jan 2025 17:57:19 +0100 Subject: [PATCH] Fix hhbtag score handling (@taaaeen). --- hbt/production/hhbtag.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/hbt/production/hhbtag.py b/hbt/production/hhbtag.py index 5e08ff73..cdf4ba8b 100644 --- a/hbt/production/hhbtag.py +++ b/hbt/production/hhbtag.py @@ -99,11 +99,11 @@ def split(where): even_mask = ak.to_numpy((events[event_mask].event % 2) == 0) if ak.sum(even_mask): input_features_even = split(even_mask) - scores_even = self.hhbtag_model_even(input_features_even)[0].numpy() + scores_even = self.hhbtag_model_even(input_features_even).numpy() scores[even_mask] = scores_even if ak.sum(~even_mask): input_features_odd = split(~even_mask) - scores_odd = self.hhbtag_model_odd(input_features_odd)[0].numpy() + scores_odd = self.hhbtag_model_odd(input_features_odd).numpy() scores[~even_mask] = scores_odd # remove the scores of padded jets