From 9a1c01cd26804f56ce2439a1658d5397471f993a Mon Sep 17 00:00:00 2001 From: Heberto Mayorquin Date: Fri, 13 Dec 2024 09:48:31 -0600 Subject: [PATCH] fix more typing errors --- src/ibl_to_nwb/datainterfaces/_pupil_tracking.py | 3 ++- src/ibl_to_nwb/datainterfaces/_roi_motion_energy.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) 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