159dc24fec0fba88f0524ef7aded7b60b12f3cc5,nussl/separation/ft2d.py,FT2D,run,#FT2D#,42

Before Change

            background_stft.append(stft_with_mask)

        background_stft = np.array(background_stft).transpose((1, 2, 0))
        self.background = AudioSignal(stft=background_stft,
                                      sample_rate=self.audio_signal.sample_rate)
        self.background.istft(self.stft_params.window_length, self.stft_params.hop_length,
                              self.stft_params.window_type,
                              overwrite=True, use_librosa=self.use_librosa_stft,
                              truncate_to_length=self.audio_signal.signal_length)

After Change

        background_masks = []
        foreground_masks = []
        for ch in range(self.audio_signal.num_channels):
            background_mask, foreground_mask = self.compute_ft2d_mask(self.ft2d, ch)
            background_mask[0:self.high_pass_cutoff, :] = 1  // high-pass filter the foreground
            background_masks.append(background_mask)
            foreground_masks.append(foreground_mask)

Instances