Tweaks

nevis/_interpolation.py

@@ -44,7 +44,7 @@ class linear_interpolant(object):
     def __init__(self, grad=False):
         self._heights = nevis.gb()
         self._resolution = nevis.spacing()
-        self.grad = grad
+        self._grad = grad
 
     def __call__(self, x, y):
         ny, nx = self._heights.shape
@@ -78,7 +78,7 @@ def __call__(self, x, y):
 
         # Final result
         f = float(np.where(f1 == f2, f1, (y2 - y) * f1 + (y - y1) * f2))
-        if self.grad:
+        if self._grad:
             # Gradient of the interpolant
             g = (h21 - h11) / self._resolution, (h12 - h11) / self._resolution
             return f, g

nevis/_util.py

@@ -4,9 +4,6 @@
 """
 import timeit
 
-from pykml.factory import KML_ElementMaker as KML
-from lxml import etree
-
 import nevis
 
 
@@ -146,6 +143,8 @@ def generate_kml(path, labels=None, trajectory=None, points=None,):
 
     ``lables``, ``trajectory``, and ``points`` can be used simultaneously.
     """
+    from pykml.factory import KML_ElementMaker as KML
+    from lxml import etree
 
     def make_coords(x, y):
         """