xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-11 23:45:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

Annotate content_types as Literal["image", "video", "audio"]

Browse Source
This commit is contained in:
christian-byrne 2024-07-19 15:18:01 -07:00
parent c83fb8a9c2
commit 20815cbb56
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
folder_paths.py
View File

@ -2,7 +2,7 @@ import os
import time import time
import mimetypes import mimetypes
import logging import logging
from typing import Set, List, Dict, Tuple from typing import Set, List, Dict, Tuple, Literal
supported_pt_extensions: Set[str] = set(['.ckpt', '.pt', '.bin', '.pth', '.safetensors', '.pkl']) supported_pt_extensions: Set[str] = set(['.ckpt', '.pt', '.bin', '.pth', '.safetensors', '.pkl'])
@ -89,7 +89,7 @@ def get_directory_by_type(type_name):
return get_input_directory() return get_input_directory()
return None return None
def filter_files_content_types(files: List[str], content_types: List[str]) -> List[str]: def filter_files_content_types(files: List[str], content_types: Literal["image", "video", "audio"]) -> List[str]:
""" """
Example: Example:
files = os.listdir(folder_paths.get_input_directory()) files = os.listdir(folder_paths.get_input_directory())