Fixed contours levels integer kwarg

holoviz · Oct 22, 2018 · 0147b3e · 0147b3e
1 parent 026d700
commit 0147b3e
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/holoviews/operation/element.py b/holoviews/operation/element.py
@@ -441,14 +441,18 @@ def _process(self, element, key=None):
 
         levels = self.p.levels
         zmin, zmax = element.range(2)
-        if zmin == zmax:
-            contours = contour_type([], [xdim, ydim], vdims)
-            return (element * contours) if self.p.overlaid else contours
+        if isinstance(self.p.levels, int):
+            if zmin == zmax:
+                contours = contour_type([], [xdim, ydim], vdims)
+                return (element * contours) if self.p.overlaid else contours
+            kwargs = {'N': self.p.levels}
+        else:
+            kwargs = {'levels': levels}
 
         fig = Figure()
         ax = Axes(fig, [0, 0, 1, 1])
         contour_set = QuadContourSet(ax, *data, filled=self.p.filled,
-                                     extent=extent, levels=levels)
+                                     extent=extent, **kwargs)
         levels = np.array(contour_set.get_array())
         crange = levels.min(), levels.max()
         if self.p.filled: