def __init__(
        self,
        n_fft: int,
        hop_length: int,
        center: bool = True,
        onesided: bool = True,
        return_complex: bool = True,
        normalized: bool = True,
    ):
        """
        Parameters
        ----------
        n_fft : int
            Number of FFT points (window size).
        steps : int
            Number of hops per window (i.e. hop_length = n_fft // steps).
        """
        super().__init__()
        self.n_fft = n_fft
        self.hop_length = hop_length
        self.center = center
        self.one_sided = onesided
        self.return_complex = return_complex
        self.normalized = normalized

        Parameters
        ----------
        n_fft : int
            Number of FFT points (window size).
        steps : int
            Number of hops per window (i.e. hop_length = n_fft // steps).
        
braindecode/models/contrawr.py