TokaMaker: Fix matplotlib exception in "plot_eddy" method

hansec · Feb 26, 2024 · 70dfee4 · 70dfee4
1 parent 69b2704
commit 70dfee4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/python/OpenFUSIONToolkit/TokaMaker.py b/src/python/OpenFUSIONToolkit/TokaMaker.py
@@ -1291,7 +1291,7 @@ def plot_eddy(self,fig,ax,dpsi_dt=None,nlevels=40,colormap='jet',clabel=r'$J_w$
                 field_tmp = dpsi_dt/eta
                 mesh_currents[mask_tmp] = numpy.sum(field_tmp[self.lc[mask_tmp,:]],axis=1)/3.0
                 mask = numpy.logical_or(mask,mask_tmp)
-        clf = ax.tripcolor(self.r[:,0],self.r[:,1],self.lc[mask],mesh_currents[mask],nlevels,cmap=colormap)
+        clf = ax.tripcolor(self.r[:,0],self.r[:,1],self.lc[mask],mesh_currents[mask],cmap=colormap)
         cb = fig.colorbar(clf,ax=ax)
         cb.set_label(clabel)
         # Make 1:1 aspect ratio