update track label

RasmusOrsoe · Apr 30, 2024 · f513072 · f513072
1 parent 9464f7a
commit f513072
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/graphnet/training/labels.py b/src/graphnet/training/labels.py
@@ -103,4 +103,4 @@ def __init__(
     def __call__(self, graph: Data) -> torch.tensor:
         """Compute label for `graph`."""
         label = (graph[self._pid_key] == 14) & (graph[self._int_key] == 1)
-        return torch.tensor(label, dtype=torch.int64)
+        return label  # torch.tensor(label, dtype=torch.int64)