properly connected cmap

mtlam · Aug 4, 2020 · 255b28f · 255b28f
1 parent 3549f41
commit 255b28f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pypulse/utils.py b/pypulse/utils.py
@@ -227,9 +227,9 @@ def imshow(x, ax=None, origin='lower', interpolation='nearest', aspect='auto', c
             cmap = plt.cm.viridis
 
     if ax is not None:
-        im = ax.imshow(x, origin=origin, interpolation=interpolation, aspect=aspect, **kwargs)
+        im = ax.imshow(x, origin=origin, interpolation=interpolation, aspect=aspect, cmap=cmap, **kwargs)
     else:
-        im = plt.imshow(x, origin=origin, interpolation=interpolation, aspect=aspect, **kwargs)
+        im = plt.imshow(x, origin=origin, interpolation=interpolation, aspect=aspect, cmap=cmap, **kwargs)
     return im
 
 '''