Skip to content

Commit

Permalink
FIX: Ruff flakes
Browse files Browse the repository at this point in the history
  • Loading branch information
scott-huberty committed Jan 20, 2024
1 parent 4159996 commit ffff381
Show file tree
Hide file tree
Showing 3 changed files with 13 additions and 19 deletions.
4 changes: 1 addition & 3 deletions eoglearn/io/eegeyenet.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,7 @@
from importlib import import_module

import numpy as np

import mne
from mne._fiff.constants import FIFF
import numpy as np


def _check_pymatreader_installed():
Expand Down
1 change: 0 additions & 1 deletion eoglearn/io/tests/test_io.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,3 @@
import numpy as np

from mne._fiff.constants import FIFF

Expand Down
27 changes: 12 additions & 15 deletions eoglearn/models/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,6 @@
import matplotlib.pyplot as plt
import mne
import numpy as np
import mne
from mne.utils import logger
from sklearn.preprocessing import StandardScaler
from tensorflow.keras.layers import LSTM
Expand All @@ -19,28 +18,26 @@
class EOGDenoiser:
"""Use simultaneous EEG and Eyetracking to Denoise EOG from the EEG data.
Parameters
----------
Parameters
----------
raw : mne.io.Raw
An instance of ``mne.io.Raw``, with EEG, eyegaze, and pupil channels.
downsample : int
The factor by which to downsample the EEG and eyetracking data. EEG channels
will be low-pass filtered before downsampling using ``mne.io.filter.resample``.
Eyetracking channels will be decimated without any filtering. resampling and
decimating will be done on copies of the data, so the original input data will
be preserved.
will be low-pass filtered before downsampling using
``mne.io.filter.resample``. Eyetracking channels will be decimated without
any filtering. resampling and decimating will be done on copies of the data,
so the original input data will be preserved.
n_units : int
The number of units to pass into the initial LSTM layer. Defaults to 50.
n_times : int
The number of timepoints to pass into the LSTM model at once. Defaults to 100.
The number of timepoints to pass into the LSTM model at once. Defaults to
100.
noise_picks: list | None
Channels that contain the noise channels.
<<<<<<< HEAD
=======
>>>>>>> lina/main
Attributes
----------
Attributes
----------
raw : mne.io.Raw
The original input ``mne.io.Raw`` instance.
downsample : int
Expand Down Expand Up @@ -72,8 +69,8 @@ class EOGDenoiser:
i.e. ``(n_samples, n_meeg_channels)`` of the input :class:`~mne.io.Raw`
object.
Notes
-----
Notes
-----
See the MNE-Python tutorial on aligning EEG and eyetracking data for information
on how to create a raw object with both EEG and eyetracking channels.
"""
Expand Down

0 comments on commit ffff381

Please sign in to comment.