diff --git a/src/ibl_to_nwb/datainterfaces/_pupil_tracking.py b/src/ibl_to_nwb/datainterfaces/_pupil_tracking.py index fb7b626..9b4901f 100644 --- a/src/ibl_to_nwb/datainterfaces/_pupil_tracking.py +++ b/src/ibl_to_nwb/datainterfaces/_pupil_tracking.py @@ -1,6 +1,7 @@ """Data Interface for the pupil tracking.""" from pathlib import Path +from typing import Optional import numpy as np from neuroconv.basedatainterface import BaseDataInterface @@ -12,7 +13,7 @@ class PupilTrackingInterface(BaseDataInterface): - def __init__(self, one: ONE, session: str, camera_name: str, revision: str | None = None): + def __init__(self, one: ONE, session: str, camera_name: str, revision: Optional[str | None] = None): self.one = one self.session = session self.camera_name = camera_name diff --git a/src/ibl_to_nwb/datainterfaces/_roi_motion_energy.py b/src/ibl_to_nwb/datainterfaces/_roi_motion_energy.py index 8ea21d3..0a40f2e 100644 --- a/src/ibl_to_nwb/datainterfaces/_roi_motion_energy.py +++ b/src/ibl_to_nwb/datainterfaces/_roi_motion_energy.py @@ -1,4 +1,5 @@ """Data Interface for the special data type of ROI Motion Energy.""" +from typing import Optional from neuroconv.basedatainterface import BaseDataInterface from neuroconv.tools.nwb_helpers import get_module @@ -7,7 +8,7 @@ class RoiMotionEnergyInterface(BaseDataInterface): - def __init__(self, one: ONE, session: str, camera_name: str, revision: str | None = None): + def __init__(self, one: ONE, session: str, camera_name: str, revision: Optional[str] = None): self.one = one self.session = session self.camera_name = camera_name