apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

apply style

This commit is contained in:
BZLibby 2023-12-16 14:56:19 -06:00
parent 6a561173a7
commit 90bc28d06e
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/onnx_web/server/load.py
View File

@ -519,16 +519,18 @@ def load_wildcards(server: ServerContext) -> None:
parse_wildcards(data, root_key=path.splitext(file)[0]) parse_wildcards(data, root_key=path.splitext(file)[0])
def parse_wildcards(data: Any, root_key: Optional[str]=None) -> None: def parse_wildcards(data: Any, root_key: Optional[str] = None) -> None:
global wildcard_data global wildcard_data
for key, values in data.items(): for key, values in data.items():
if root_key is not None: if root_key is not None:
key=f"{root_key}{sep}{key}" key = f"{root_key}{sep}{key}"
if isinstance(values, dict): if isinstance(values, dict):
parse_wildcards(values, root_key=key) parse_wildcards(values, root_key=key)
elif isinstance(values, list): elif isinstance(values, list):
wildcard_data[key].extend(values) wildcard_data[key].extend(values)
else: else:
logger.warning("unable to parse key: %s from wildcard path: %s", key, root_key) logger.warning(
"unable to parse key: %s from wildcard path: %s", key, root_key
)