move jit import out of exception

arm61 · Dec 19, 2018 · db3d102 · db3d102
1 parent 0e927db
commit db3d102
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pylj/pairwise.py b/pylj/pairwise.py
@@ -1,12 +1,11 @@
 from __future__ import division
 import numpy as np
-
+from numba import jit
 try:
     from pylj import comp as heavy
 except ImportError:
     print("WARNING, using slow force and energy calculations")
     from pylj import pairwise as heavy
-    from numba import jit
 
 
 @jit