3.8 Compatible type annotation (#3938)

comfyanonymous · Jul 3, 2024 · 086ac75 · 086ac75
1 parent d7484ef
commit 086ac75
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/folder_paths.py b/folder_paths.py
@@ -1,12 +1,13 @@
 import os
 import time
 import logging
+from typing import Set, List, Dict, Tuple
 
-supported_pt_extensions: set[str] = set(['.ckpt', '.pt', '.bin', '.pth', '.safetensors', '.pkl'])
+supported_pt_extensions: Set[str] = set(['.ckpt', '.pt', '.bin', '.pth', '.safetensors', '.pkl'])
 
-SupportedFileExtensionsType = set[str]
-ScanPathType = list[str]
-folder_names_and_paths: dict[str, tuple[ScanPathType, SupportedFileExtensionsType]] = {}
+SupportedFileExtensionsType = Set[str]
+ScanPathType = List[str]
+folder_names_and_paths: Dict[str, Tuple[ScanPathType, SupportedFileExtensionsType]] = {}
 
 base_path = os.path.dirname(os.path.realpath(__file__))
 models_dir = os.path.join(base_path, "models")