1580158
/
stable-diffusion-webui-text2prompt
mirror of https://github.com/toshiaki1729/stable-diffusion-webui-text2prompt.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix #3 

fix type hint incompatible with python < 3.9
main
toshiaki1729 2023-02-24 17:29:55 +09:00
parent a6c052ef7a
commit c4d8cec126
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/t2p/prompt_generator/database_loader.py
View File

@ -8,11 +8,11 @@ import scripts.t2p.settings as settings
class Database:
def __init__(self, database_path: str, re_filename: re.Pattern[str]):
def __init__(self, database_path: str, re_filename: re.Pattern):
self.read_files(database_path, re_filename)
def read_files(self, database_path: str, re_filename: re.Pattern[str]):
def read_files(self, database_path: str, re_filename: re.Pattern):
self.clear()
self.database_path = database_path
fn, _ = os.path.splitext(os.path.basename(database_path))
@ -77,12 +77,12 @@ class Database:
class DatabaseLoader:
def __init__(self, path: str, re_filename: re.Pattern[str]):
def __init__(self, path: str, re_filename: re.Pattern):
self.datas: Dict[str, Database] = dict()
self.preload(path, re_filename)
def preload(self, path: str, re_filename: re.Pattern[str]):
def preload(self, path: str, re_filename: re.Pattern):
dirs = os.listdir(path)
for d in dirs:
filepath = os.path.join(path, d)