diff --git a/PyStageLinQ/Network.py b/PyStageLinQ/Network.py index b1c3442..4ce80cb 100644 --- a/PyStageLinQ/Network.py +++ b/PyStageLinQ/Network.py @@ -2,6 +2,7 @@ (c) 2022 Jaxcie This code is licensed under MIT license (see LICENSE for details) """ + from __future__ import annotations import asyncio from . import EngineServices @@ -121,11 +122,14 @@ async def _receive_data_loop(self) -> None: async def _receive_frames( self, - ) -> bool | list[ - StageLinQServiceAnnouncementData - | StageLinQReferenceData - | StageLinQServiceRequestService - ]: + ) -> ( + bool + | list[ + StageLinQServiceAnnouncementData + | StageLinQReferenceData + | StageLinQServiceRequestService + ] + ): response = await self.reader.read(1024) if len(response) == 0: # Socket closed diff --git a/PyStageLinQ/PyStageLinQ.py b/PyStageLinQ/PyStageLinQ.py index 6b30d19..a3e317a 100644 --- a/PyStageLinQ/PyStageLinQ.py +++ b/PyStageLinQ/PyStageLinQ.py @@ -2,6 +2,7 @@ (c) 2022 Jaxcie This code is licensed under MIT license (see LICENSE for details) """ + import select import socket import time diff --git a/tests/Main.py b/tests/Main.py index 6ed9878..615fffd 100644 --- a/tests/Main.py +++ b/tests/Main.py @@ -2,6 +2,7 @@ (c) 2022 Jaxcie This code is licensed under MIT license (see LICENSE for details) """ + from PyStageLinQ import EngineServices, PyStageLinQ """