Make sure to make link collections available

AIDASoft · Oct 1, 2024 · fdd6598 · fdd6598
1 parent f89ff67
commit fdd6598
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/podio_gen/cpp_generator.py b/python/podio_gen/cpp_generator.py
@@ -525,8 +525,10 @@ def _write_list(name, target_folder, files, comment):
 
     def _write_all_collections_header(self):
         """Write a header file that includes all collection headers"""
-
-        collection_files = (x.split("::")[-1] + "Collection.h" for x in self.datamodel.datatypes)
+        collection_files = (
+            x.split("::")[-1] + "Collection.h"
+            for x in list(self.datamodel.datatypes.keys()) + list(self.datamodel.links.keys())
+        )
         self._write_file(
             os.path.join(self.install_dir, self.package_name, f"{self.package_name}.h"),
             self._eval_template(