lbne model now imports python modules directly

1 files changed, 13 insertions, 6 deletions

diff --git a/detectors/lbne.py b/detectors/lbne.py
index 7cd666a..ff38ef4 100644
--- a/detectors/lbne.py
+++ b/detectors/lbne.py
@@ -1,13 +1,20 @@
-import sys
 import numpy as np
 from copy import deepcopy
-from chroma import layout
-from chroma.stl import read_stl
-from chroma.transform import rotate
-from chroma.geometry import Geometry, Solid
-from chroma.materials import glass, h2o
 from itertools import product
 
+import os
+import sys
+
+dir = os.path.split(os.path.realpath(__file__))[0]
+
+sys.path.append(dir + '/..')
+
+import layout
+from stl import read_stl
+from transform import rotate
+from geometry import Geometry, Solid
+from materials import glass, h2o
+
 endcap_spacing = .485
 
 radius = 25.0/10.0