3.8 Compatible type annotation (#3938)

086ac752 · Chenlei Hu · GitHub · d7484ef3 · 086ac752
Unverified Commit 086ac752 authored Jul 03, 2024 by Chenlei Hu Committed by GitHub Jul 03, 2024
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 4 deletions

folder_paths.py folder_paths.py +5 -4

No files found.
--- a/folder_paths.py
+++ b/folder_paths.py
 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")