From e2943ba56e1d244d440791200ae460cd7b72aa0e Mon Sep 17 00:00:00 2001 From: rowanc1 Date: Tue, 26 Nov 2013 17:31:42 -0800 Subject: [PATCH] Bug fix if no colorer is used in the video. --- SimPEG/mesh/TensorView.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/SimPEG/mesh/TensorView.py b/SimPEG/mesh/TensorView.py index 4b1c4fc3..9956565d 100644 --- a/SimPEG/mesh/TensorView.py +++ b/SimPEG/mesh/TensorView.py @@ -372,7 +372,7 @@ class TensorView(object): return animate(fig, animateFrame, frames=mesh.nCv['xyz'.index(normal)]) - def video(mesh,var,function,figsize=(10,8)): + def video(mesh,var,function,figsize=(10,8),colorbar=True): """ Call a function for a list of models to create a video. @@ -390,7 +390,8 @@ class TensorView(object): VAR = np.concatenate(var) clim = [VAR.min(),VAR.max()] tlt = plt.title('') - plt.colorbar(function(var[0],ax,clim,tlt,0)) + if colorbar: + plt.colorbar(function(var[0],ax,clim,tlt,0)) def animateFrame(i): function(var[i],ax,clim,tlt,i)