Minor change of allowing an optional "display" argument to the plots …

…script.
tnowotny · Feb 6, 2021 · ffe6135 · ffe6135
1 parent c40c690
commit ffe6135
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/exp1_plots.py b/src/exp1_plots.py
@@ -2,7 +2,7 @@
 import numpy as np
 from helper import *
 
-def exp1_plots(state_bufs, spike_t, spike_ID, plot_raster, plot_sdf, t_total, dt, n_glo, n, N, dirname, label):
+def exp1_plots(state_bufs, spike_t, spike_ID, plot_raster, plot_sdf, t_total, dt, n_glo, n, N, dirname, label,display= True):
     dirname= dirname+"/"
     dt_sdf= 1.0
     sigma_sdf= 300.0
@@ -65,4 +65,5 @@ def exp1_plots(state_bufs, spike_t, spike_ID, plot_raster, plot_sdf, t_total, dt
         plt.savefig(dirname+label+"_"+pop+"_sdftraces.png",dpi=300)
 
     # Show plot
-    plt.show()
+    if display:
+        plt.show()