From 9d65a63726da9fbcd9d6b823fd75a94819c15590 Mon Sep 17 00:00:00 2001
From: jmcarcell <jmcarcell@users.noreply.github.com>
Date: Mon, 2 Sep 2024 08:27:56 +0200
Subject: [PATCH] Add missing collections to the frames

---
 scripts/createEDM4hepFile.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/scripts/createEDM4hepFile.py b/scripts/createEDM4hepFile.py
index 9fae31fcc..d8e070fe2 100644
--- 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")