diff --git a/python/edm4hep/__init__.py b/python/edm4hep/__init__.py index 1cf2df10d..205f4749c 100644 --- a/python/edm4hep/__init__.py +++ b/python/edm4hep/__init__.py @@ -24,6 +24,9 @@ raise RuntimeError('Failed to load Constants.h') from ROOT import edm4hep +from podio.pythonizations import load_pythonizations +load_pythonizations('edm4hep') + # Make TAB completion work for utils setattr(edm4hep, 'utils', edm4hep.utils) diff --git a/test/test_EDM4hepFile.py b/test/test_EDM4hepFile.py index fc8da2d14..51f0445ad 100644 --- a/test/test_EDM4hepFile.py +++ b/test/test_EDM4hepFile.py @@ -250,7 +250,6 @@ def test_ClusterCollection(event): calo_hit = event.get("CalorimeterHitCollection")[0] assert len(cluster.getHits()) == 1 assert cluster.getHits()[0] == calo_hit - cluster.addToHits(calo_hit) def test_TrackerHit3DCollection(event):