fix: use typing module to fix dep for py 3.6

font_dataset/font.py

@@ -1,5 +1,7 @@
 import yaml
 import os
+from typing import List
+
 
 from .utils import get_files
 
@@ -13,7 +15,7 @@ class DSFont:
         self.language = language
 
 
-def load_fonts(config_path="configs/font.yml") -> list[DSFont]:
+def load_fonts(config_path="configs/font.yml") -> List[DSFont]:
     with open(config_path, "r", encoding="utf-8") as f:
         config = yaml.safe_load(f)
 

font_dataset/fontlabel.py

@@ -1,5 +1,6 @@
 __all__ = ["FontLabel"]
 
+from typing import Tuple
 from .font import DSFont
 
 
@@ -26,14 +27,14 @@ class FontLabel:
         image_height: int,
         text: str,
         font: DSFont,
-        text_color: tuple[int, int, int],
+        text_color: Tuple[int, int, int],
         text_size: int,
         text_direction: str,
         stroke_width: int,
-        stroke_color: tuple[int, int, int],
+        stroke_color: Tuple[int, int, int],
         line_spacing: int,
         language: str,
-        bbox: tuple[int, int, int, int],
+        bbox: Tuple[int, int, int, int],
         angle: int,
     ):
         self.image_width = image_width

font_dataset/layout.py

@@ -1,3 +1,5 @@
+from typing import Tuple
+
 __all__ = ["generate_font_image"]
 
 
@@ -237,7 +239,7 @@ def RGB2RGBA(color):
 
 def generate_font_image(
     img_path: str, font: DSFont, corpus_manager: CorpusGeneratorManager
-) -> tuple[Image.Image, FontLabel]:
+) -> Tuple[Image.Image, FontLabel]:
     im = Image.open(img_path)
     # crop image
     width, height = im.size