fixing type hint

cvat-ai · Jul 11, 2024 · 4257150 · 4257150
1 parent 6adf992
commit 4257150
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/datumaro/plugins/datumaro_format/extractor.py b/datumaro/plugins/datumaro_format/extractor.py
@@ -3,6 +3,7 @@
 # SPDX-License-Identifier: MIT
 
 import os.path as osp
+from typing import List, Union
 
 from datumaro.components.annotation import (
     AnnotationType,
@@ -269,8 +270,8 @@ def _load_annotations(self, item: dict):
         return loaded
 
     def _load_skeleton_elements_annotations(
-        self, ann: dict, label_id: int, points: list[float | int]
-    ) -> list[Points]:
+        self, ann: dict, label_id: int, points: List[Union[float, int]]
+    ) -> List[Points]:
         if len(points) % 3 != 0:
             raise InvalidAnnotationError(
                 f"Points have invalid value count {len(points)}, "