Early return on plotting code.

simpeg · Oct 30, 2015 · cd01f07 · cd01f07
1 parent 465087b
commit cd01f07
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/SimPEG/Examples/DCfwd.py b/SimPEG/Examples/DCfwd.py
@@ -38,6 +38,7 @@ def DCfun(mesh, pts):
     AinvrM = SolverLU(ArM)
     phirM = AinvrM*rhsrM
 
+    if not plotIt: return
     #Step4: Making Figure
     fig, axes = plt.subplots(1,2,figsize=(12*1.2,4*1.2))
     label = ["(a)", "(b)"]
@@ -69,7 +70,6 @@ def DCfun(mesh, pts):
             axes[i].set_ylabel(" ")
         axes[i].set_xlabel("x")
 
-    if plotIt: plt.show()
 
 if __name__ == '__main__':
     run()
diff --git a/SimPEG/FLOW/Examples/Celia1990.py b/SimPEG/FLOW/Examples/Celia1990.py
@@ -27,6 +27,7 @@ def getFields(timeStep,method):
     Hs_H30 = getFields(30., 'head')
     Hs_H120= getFields(120.,'head')
 
+    if not plotIt:return
     plt.figure(figsize=(13,5))
     plt.subplot(121)
     plt.plot(40-M.gridCC, Hs_M10[-1],'b-')
@@ -46,7 +47,6 @@ def getFields(timeStep,method):
     plt.xlabel('Depth, cm')
     plt.ylabel('Pressure Head, cm')
     plt.legend(('$\Delta t$ = 10 sec','$\Delta t$ = 30 sec','$\Delta t$ = 120 sec'))
-    if plotIt:plt.show()
 
 if __name__ == '__main__':
     run()