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