bugfix, modules io,lfp. tuple to Tuple

ryanharvey1 · Oct 16, 2024 · d2930f9 · d2930f9
1 parent 1bda637
commit d2930f9
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/neuro_py/io/loading.py b/neuro_py/io/loading.py
@@ -2240,7 +2240,7 @@ def load_probe_layout(basepath: str) -> pd.DataFrame:
     return probe_layout
 
 
-def load_emg(basepath: str, threshold: float = 0.9) -> tuple[nel.AnalogSignalArray, nel.EpochArray, nel.EpochArray]:
+def load_emg(basepath: str, threshold: float = 0.9) -> Tuple[nel.AnalogSignalArray, nel.EpochArray, nel.EpochArray]:
     """
     Load EMG data from basename.EMGFromLFP.LFP.mat.
 

diff --git a/neuro_py/lfp/preprocessing.py b/neuro_py/lfp/preprocessing.py
@@ -1,3 +1,4 @@
+from typing import Tuple
 import nelpy as nel
 import numpy as np
 
@@ -6,8 +7,8 @@
 
 def clean_lfp(
     lfp: nel.AnalogSignalArray,
-    thresholds: tuple[float, float] = (5, 10),
-    artifact_time_expand: tuple[float, float] = (0.25, 0.1),
+    thresholds: Tuple[float, float] = (5, 10),
+    artifact_time_expand: Tuple[float, float] = (0.25, 0.1),
 ) -> np.ndarray:
     """
     Remove artefacts and noise from a local field potential (LFP) signal.

diff --git a/neuro_py/lfp/spectral.py b/neuro_py/lfp/spectral.py
@@ -1,4 +1,5 @@
 from concurrent.futures import ThreadPoolExecutor
+from typing import Tuple
 
 import numpy as np
 import pandas as pd
@@ -47,7 +48,7 @@ def event_triggered_wavelet(
     parallel: bool = True,
     whiten: bool = True,
     whiten_order: int = 2,
-) -> tuple[np.ndarray, np.ndarray, np.ndarray, np.ndarray] | tuple[pd.DataFrame, pd.Series]:
+) -> Tuple[np.ndarray, np.ndarray, np.ndarray, np.ndarray] | Tuple[pd.DataFrame, pd.Series]:
     """
     Compute the event-triggered wavelet transform of a signal.