diff --git a/src/ibl_to_nwb/datainterfaces/_brainwide_map_trials.py b/src/ibl_to_nwb/datainterfaces/_brainwide_map_trials.py index c7594f7..e82b548 100644 --- 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 index 375f854..f94f71d 100644 --- 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 index 00ec234..f5420f5 100644 --- 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