TypedDict from typing_extensions

mio/models/pipeline.py

@@ -6,18 +6,22 @@
 from abc import abstractmethod
 from datetime import datetime
 from graphlib import TopologicalSorter
-from typing import Any, ClassVar, Generic, Optional, TypedDict, TypeVar, Union, Unpack, final
+from typing import Any, ClassVar, Generic, Optional, TypeVar, Union, Unpack, final
 
 from pydantic import Field, field_validator, model_validator
 
 from mio.exceptions import ConfigurationMismatchError
 from mio.models.models import MiniscopeConfig, PipelineModel
 
 if sys.version_info < (3, 11):
-    from typing_extensions import Self
-else:
+    from typing_extensions import Self, TypedDict
+elif sys.version_info < (3, 12):
     from typing import Self
 
+    from typing_extensions import TypedDict
+else:
+    from typing import Self, TypedDict
+
 T = TypeVar("T")
 """
 Input Type typevar

mio/sinks/return_.py

@@ -2,10 +2,16 @@
 Special Return sink that pipeline runners use to return values from :meth:`.PipelineRunner.process`
 """
 
-from typing import Any, Optional, TypedDict
+import sys
+from typing import Any, Optional
 
 from mio.models.pipeline import Sink, T
 
+if sys.version_info < (3, 12):
+    from typing_extensions import TypedDict
+else:
+    from typing import TypedDict
+
 
 class ReturnConfig(TypedDict):
     """

mio/sources/file.py

@@ -1,9 +1,9 @@
 """
 File-based data sources
 """
-
+import sys
 from pathlib import Path
-from typing import BinaryIO, ClassVar, Optional, TypedDict
+from typing import BinaryIO, ClassVar, Optional
 
 import numpy as np
 from pydantic import Field
@@ -12,6 +12,11 @@
 from mio.models.pipeline import Source
 from mio.models.sdcard import SDBufferHeader, SDConfig, SDLayout
 
+if sys.version_info < (3, 12):
+    from typing_extensions import TypedDict
+else:
+    from typing import TypedDict
+
 
 class FileSource(Source):
     """

mio/transforms/frame.py

@@ -1,15 +1,20 @@
 """
 Nodes that receive and emit frames
 """
-
-from typing import Optional, TypedDict
+import sys
+from typing import Optional
 
 import numpy as np
 
 from mio.models.buffer import BufferHeader
 from mio.models.data import Frame
 from mio.models.pipeline import Transform
 
+if sys.version_info < (3, 12):
+    from typing_extensions import TypedDict
+else:
+    from typing import TypedDict
+
 
 class MergeBuffersConfig(TypedDict):
     """Configuration for :class:`.MergeBuffers`"""