Merge pull request #52 from vhewes/bugfix/3d-vertex

add second dimension to 3D vertex truth
nugraph · Aug 24, 2023 · 7dc5c7b · 7dc5c7b
2 parents 1a25e42 + be82c4f
commit 7dc5c7b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pynuml/process/hitgraph.py b/pynuml/process/hitgraph.py
@@ -180,7 +180,7 @@ def __call__(self, evt: 'pynuml.io.Event') -> tuple[str, Any]:
 
         # 3D vertex truth
         if self.label_vertex:
-            vtx_3d = [ event.nu_vtx_x, event.nu_vtx_y, event.nu_vtx_z ]
+            vtx_3d = [ [ event.nu_vtx_x, event.nu_vtx_y, event.nu_vtx_z ] ]
             data['evt'].y_vtx = torch.tensor(vtx_3d).float()
 
         return evt.name, data