small_fix

graphnet-team · Dec 6, 2024 · 75b3260 · 75b3260
1 parent 165fedb
commit 75b3260
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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