diff --git a/signalgen.py b/signalgen.py index ea187e9..a0b41e5 100755 --- a/signalgen.py +++ b/signalgen.py @@ -88,7 +88,7 @@ def addBurst(values): spectra = filterbankObj.get_data(0, samples) #save pre-injection spectra plot - if not skipplot: + if not values.skipplot: show_dynamic(spectra, f"{basename} Pre-injection Dynamic Spectra", save=True) #get bandpass and store in bpWeights @@ -123,7 +123,7 @@ def addBurst(values): ) #and save the new plot - if not skipplot: + if not values.skipplot: logger.info("Saving plot...") show_dynamic(injectedSpectra, f"{basename} Dynamic Spectra and {values.dm} DM Pulse", save=True) #now generate new filterbank file