rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-03-29 09:50:05 +01:00
Code Issues Projects Releases Wiki Activity

Fix the type hint for bridged_kwarg decorator

Browse Source
This commit is contained in:
Gnome Ann 2022-01-04 20:48:34 -05:00
parent fc6caa0df0
commit 01479c29ea
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
aiserver.py
View File

@ -21,7 +21,7 @@ import zipfile
import packaging import packaging
import contextlib import contextlib
import traceback import traceback
from typing import Any, Callable, Union, Dict, Set, List from typing import Any, Callable, TypeVar, Union, Dict, Set, List
import requests import requests
import html import html
@ -1065,8 +1065,9 @@ def lua_log_format_name(name):
return f"[{name}]" if type(name) is str else "CORE" return f"[{name}]" if type(name) is str else "CORE"
_bridged = {} _bridged = {}
F = TypeVar("F", bound=Callable)
def bridged_kwarg(name=None): def bridged_kwarg(name=None):
def _bridged_kwarg(f: Callable): def _bridged_kwarg(f: F):
_bridged[name if name is not None else f.__name__[4:] if f.__name__[:4] == "lua_" else f.__name__] = f _bridged[name if name is not None else f.__name__[4:] if f.__name__[:4] == "lua_" else f.__name__] = f
return f return f
return _bridged_kwarg return _bridged_kwarg