Add missing collections to the frames

key4hep · Sep 2, 2024 · 9d65a63 · 9d65a63
1 parent 4ffdb36
commit 9d65a63
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/scripts/createEDM4hepFile.py b/scripts/createEDM4hepFile.py
@@ -344,6 +344,7 @@
     gep.setAlphaQCD(next(counter))
     gep.setSignalProcessId(next(counter))
     gep.setSqrts(next(counter))
+    frame.put(gep_coll, "GeneratorEventParametersCollection")
 
     for i in range(vectorsize):
         gep.addToCrossSections(next(counter))
@@ -369,5 +370,6 @@
     gpi.setXf(0, next(counter))
     gpi.setXf(1, next(counter))
     gpi.setScale(next(counter))
+    frame.put(gpi_coll, "GeneratorPdfInfoCollection")
 
     writer.write_frame(frame, "events")