add LightningPoseConverter to WideFieldNWBConverter

catalystneuro · Dec 3, 2023 · 4b01a2c · 4b01a2c
1 parent 0dc57ad
commit 4b01a2c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/pinto_lab_to_nwb/widefield/widefieldnwbconverter.py b/src/pinto_lab_to_nwb/widefield/widefieldnwbconverter.py
@@ -5,9 +5,9 @@
 import numpy as np
 from natsort import natsorted
 from neuroconv import NWBConverter
+from neuroconv.converters import LightningPoseConverter
 from pynwb import NWBFile
 
-from pinto_lab_to_nwb.widefield.interfaces import WidefieldImagingInterface, WidefieldProcessedImagingInterface
 from pinto_lab_to_nwb.widefield.utils import load_motion_correction_data
 from pinto_lab_to_nwb.widefield.utils.motion_correction import add_motion_correction
 from pinto_lab_to_nwb.widefield.interfaces import (
@@ -30,6 +30,7 @@ class WideFieldNWBConverter(NWBConverter):
         SegmentationProcessedBlue=WidefieldProcessedSegmentationinterface,
         SummaryImagesBlue=WidefieldSegmentationImagesBlueInterface,
         SummaryImagesViolet=WidefieldSegmentationImagesVioletInterface,
+        EyeTracking=LightningPoseConverter,
     )
 
     def __init__(self, source_data: Dict[str, dict], verbose: bool = True):