check

graphnet-team · May 20, 2024 · f51bcc6 · f51bcc6
1 parent b31aa42
commit f51bcc6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/graphnet/training/labels.py b/src/graphnet/training/labels.py
@@ -102,5 +102,7 @@ def __init__(
 
     def __call__(self, graph: Data) -> torch.tensor:
         """Compute label for `graph`."""
-        label = (torch.abs(graph[self._pid_key]) == 14) & (graph[self._int_key] == 1)
+        is_numu = torch.abs(graph[self._pid_key]) == 14
+        is_cc = graph[self._int_key] == 1
+        label = is_numu & is_cc
         return label.type(torch.int)