Rename pre_energy/post_energy columns to pre_E/post_E

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-06-17 14:31:41 +02:00
parent 877f83a5cd
commit d402cdace3
3 changed files with 7 additions and 7 deletions
@@ -296,7 +296,7 @@ def predict(
                "pre_x":        chunk["pre_pos"][:, 0],
                "pre_y":        chunk["pre_pos"][:, 1],
                "pre_z":        chunk["pre_pos"][:, 2],
-                "pre_energy":   chunk["pre_energy"],
+                "pre_E":   chunk["pre_E"],
                "pre_dir_x":    chunk["pre_dir"][:, 0],
                "pre_dir_y":    chunk["pre_dir"][:, 1],
                "pre_dir_z":    chunk["pre_dir"][:, 2],