Fix column names to match actual parquet schema

material_id → material, n_secondaries derived from child_track_ids.list.len() Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-06-17 14:26:57 +02:00
parent 2e4b4d91d1
commit 78c2a61ecd
2 changed files with 10 additions and 10 deletions
@@ -300,9 +300,9 @@ def predict(
                "pre_dir_x":    chunk["pre_dir"][:, 0],
                "pre_dir_y":    chunk["pre_dir"][:, 1],
                "pre_dir_z":    chunk["pre_dir"][:, 2],
-                "material_id":  chunk["material"],
+                "material":     chunk["material"],
                "layer_id":     chunk["layer_id"],
-                "n_secondaries": chunk["n_sec"],
+                "n_sec":         chunk["n_sec"],
                "step_length":  step_length,
                "delta_e":      delta_e,
                "edep":         edep,