summaryrefslogtreecommitdiff
path: root/generator
diff options
context:
space:
mode:
Diffstat (limited to 'generator')
-rw-r--r--generator/g4gen.py2
-rw-r--r--generator/vertex.py4
2 files changed, 3 insertions, 3 deletions
diff --git a/generator/g4gen.py b/generator/g4gen.py
index cec53f7..b43ce92 100644
--- a/generator/g4gen.py
+++ b/generator/g4gen.py
@@ -4,7 +4,7 @@ import g4py.NISTmaterials
import g4py.ParticleGun
import pyublas
import numpy as np
-import event
+import chroma.event as event
try:
import G4chroma
diff --git a/generator/vertex.py b/generator/vertex.py
index 1fa61cb..3998e6e 100644
--- a/generator/vertex.py
+++ b/generator/vertex.py
@@ -3,8 +3,8 @@ import numpy as np
from math import sin, cos, sqrt
import copy
-import pi0
-from event import Event, Subtrack
+import chroma.pi0 as pi0
+from chroma.event import Event, Subtrack
# generator parts for use with gun()