1 files changed, 4 insertions, 10 deletions
diff --git a/solids/pmts.py b/solids/pmts.py
index fcbc8fc..43aed3f 100644
--- a/solids/pmts.py
+++ b/solids/pmts.py
@@ -1,14 +1,8 @@
-import os
-import sys
-
-dir = os.path.split(os.path.realpath(__file__))[0]
-sys.path.append(dir + '/..')
-
 import numpy as np
-from geometry import Solid
-from make import rotate_extrude
-from optics import *
-from tools import read_csv, offset
+from chroma.geometry import Solid
+from chroma.make import rotate_extrude
+from chroma.optics import *
+from chroma.tools import read_csv, offset
 
 def get_lc_profile(radii, a, b, d, rmin, rmax):
     c = -b*np.sqrt(1 - (rmin-d)**2/a**2)