From 75b3260f830feff65c40dbff425e126e35724257 Mon Sep 17 00:00:00 2001 From: "askerosted@gmail.com" Date: Fri, 6 Dec 2024 21:51:16 +0900 Subject: [PATCH] small_fix --- src/graphnet/models/graphs/utils.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/graphnet/models/graphs/utils.py b/src/graphnet/models/graphs/utils.py index 9c9a76062..7bb746508 100644 --- a/src/graphnet/models/graphs/utils.py +++ b/src/graphnet/models/graphs/utils.py @@ -431,8 +431,9 @@ def add_percentile_summary( def add_counts(self, location: Optional[int] = None) -> np.ndarray: """Add the counts of the sensor to the summarization features.""" self._add_column(np.log10(self._counts), location) - new_name = ["counts"] - self._add_column_names(new_name, location) + if self._input_names is not None: + new_name = ["counts"] + self._add_column_names(new_name, location) def add_sum_charge( self, charge_index: int, location: Optional[int] = None