initial commit

mne/time_frequency/__init__.pyi

@@ -0,0 +1,81 @@
+__all__ = [
+    "AverageTFR",
+    "AverageTFRArray",
+    "BaseTFR",
+    "CrossSpectralDensity",
+    "EpochsSpectrum",
+    "EpochsSpectrumArray",
+    "EpochsTFR",
+    "EpochsTFRArray",
+    "RawTFR",
+    "RawTFRArray",
+    "Spectrum",
+    "SpectrumArray",
+    "csd_array_fourier",
+    "csd_array_morlet",
+    "csd_array_multitaper",
+    "csd_fourier",
+    "csd_morlet",
+    "csd_multitaper",
+    "csd_tfr",
+    "dpss_windows",
+    "fit_iir_model_raw",
+    "fwhm",
+    "istft",
+    "morlet",
+    "pick_channels_csd",
+    "psd_array_multitaper",
+    "psd_array_welch",
+    "read_csd",
+    "read_spectrum",
+    "read_tfrs",
+    "stft",
+    "stftfreq",
+    "tfr_array_morlet",
+    "tfr_array_multitaper",
+    "tfr_array_stockwell",
+    "tfr_morlet",
+    "tfr_multitaper",
+    "tfr_stockwell",
+    "write_tfrs",
+]
+from ._stft import istft, stft, stftfreq
+from ._stockwell import tfr_array_stockwell, tfr_stockwell
+from .ar import fit_iir_model_raw
+from .csd import (
+    CrossSpectralDensity,
+    csd_array_fourier,
+    csd_array_morlet,
+    csd_array_multitaper,
+    csd_fourier,
+    csd_morlet,
+    csd_multitaper,
+    csd_tfr,
+    pick_channels_csd,
+    read_csd,
+)
+from .multitaper import dpss_windows, psd_array_multitaper, tfr_array_multitaper
+from .psd import psd_array_welch
+from .spectrum import (
+    EpochsSpectrum,
+    EpochsSpectrumArray,
+    Spectrum,
+    SpectrumArray,
+    read_spectrum,
+)
+from .tfr import (
+    AverageTFR,
+    AverageTFRArray,
+    BaseTFR,
+    EpochsTFR,
+    EpochsTFRArray,
+    RawTFR,
+    RawTFRArray,
+    fwhm,
+    morlet,
+    read_tfrs,
+    tfr_array_morlet,
+    tfr_morlet,
+    tfr_multitaper,
+    write_tfrs,
+)