fix typing

catalystneuro · Dec 13, 2024 · 6c80583 · 6c80583
1 parent dab27f0
commit 6c80583
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/src/ibl_to_nwb/datainterfaces/_brainwide_map_trials.py b/src/ibl_to_nwb/datainterfaces/_brainwide_map_trials.py
@@ -1,4 +1,5 @@
 from pathlib import Path
+from typing import Optional
 
 from hdmf.common import VectorData
 from neuroconv.basedatainterface import BaseDataInterface
@@ -9,7 +10,7 @@
 
 
 class BrainwideMapTrialsInterface(BaseDataInterface):
-    def __init__(self, one: ONE, session: str, revision: str | None = None):
+    def __init__(self, one: ONE, session: str, revision: Optional[str] = None):
         self.one = one
         self.session = session
         self.revision = one.list_revisions(session)[-1] if revision is None else revision

diff --git a/src/ibl_to_nwb/datainterfaces/_lick_times.py b/src/ibl_to_nwb/datainterfaces/_lick_times.py
@@ -1,3 +1,5 @@
+from typing import Optional
+
 from hdmf.common import VectorData
 from neuroconv.basedatainterface import BaseDataInterface
 from neuroconv.tools.nwb_helpers import get_module
@@ -7,7 +9,7 @@
 
 
 class LickInterface(BaseDataInterface):
-    def __init__(self, one: ONE, session: str, revision: str | None = None):
+    def __init__(self, one: ONE, session: str, revision: Optional[str] = None):
         self.one = one
         self.session = session
         self.revision = one.list_revisions(session)[-1] if revision is None else revision

diff --git a/src/ibl_to_nwb/datainterfaces/_wheel_movement.py b/src/ibl_to_nwb/datainterfaces/_wheel_movement.py
@@ -1,4 +1,5 @@
 from pathlib import Path
+from typing import Optional
 
 from brainbox.behavior import wheel as wheel_methods
 from neuroconv.basedatainterface import BaseDataInterface
@@ -11,7 +12,7 @@
 
 
 class WheelInterface(BaseDataInterface):
-    def __init__(self, one: ONE, session: str, revision: str | None = None):
+    def __init__(self, one: ONE, session: str, revision: Optional[str] = None):
         self.one = one
         self.session = session
         self.revision = one.list_revisions(session) if revision is None else revision