def __init__(self, output, duration, dt):
self.output = output
self.duration = duration
self.dt = dt
wf = WavFile()
wf.sample_rate = 1.0 / dt
wf.data = self.output[:, 0]
wf.analyze()
self.wav_file = wf
