Use imu protobuf from pl_neon_recording

pupil-labs · Aug 2, 2024 · 96f0136 · 96f0136
1 parent df6aafa
commit 96f0136
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 491 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -28,7 +28,7 @@ install_requires =
     av
     beaupy
     numpy>=1.20
-    protobuf==3.18.0
+    pl-neon-recording
     pydantic>=2
     websockets
     zeroconf

diff --git a/src/pupil_labs/realtime_api/streaming/__init__.py b/src/pupil_labs/realtime_api/streaming/__init__.py
@@ -1,5 +1,7 @@
 import typing as T
 
+from pupil_labs.neon_recording.stream.imu.imu_pb2 import ImuPacket
+
 from .base import (
     RTSPData,
     RTSPRawStreamer,
@@ -14,7 +16,6 @@
     receive_gaze_data,
 )
 from .imu import RTSPImuStreamer, receive_imu_data
-from .imu_pb2 import ImuPacket
 from .video import RTSPVideoFrameStreamer, VideoFrame, receive_video_frames
 
 RTSPStreamerType = T.TypeVar("RTSPStreamerType", bound="RTSPRawStreamer")

diff --git a/src/pupil_labs/realtime_api/streaming/imu.py b/src/pupil_labs/realtime_api/streaming/imu.py
@@ -2,8 +2,9 @@
 import logging
 import typing as T
 
+from pupil_labs.neon_recording.stream.imu.imu_pb2 import ImuPacket
+
 from .base import RTSPRawStreamer
-from .imu_pb2 import ImuPacket
 
 logger = logging.getLogger(__name__)