apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix list parser default

This commit is contained in:
Sean Sube 2023-11-24 23:02:22 -06:00
parent 9e930a91d5
commit 17e25a3bc3
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
api/onnx_web/server/context.py
View File

@ -106,19 +106,19 @@ class ServerContext:
model_path=environ.get("ONNX_WEB_MODEL_PATH", path.join("..", "models")), model_path=environ.get("ONNX_WEB_MODEL_PATH", path.join("..", "models")),
output_path=environ.get("ONNX_WEB_OUTPUT_PATH", path.join("..", "outputs")), output_path=environ.get("ONNX_WEB_OUTPUT_PATH", path.join("..", "outputs")),
params_path=environ.get("ONNX_WEB_PARAMS_PATH", "."), params_path=environ.get("ONNX_WEB_PARAMS_PATH", "."),
cors_origin=get_list(environ, "ONNX_WEB_CORS_ORIGIN", "*"), cors_origin=get_list(environ, "ONNX_WEB_CORS_ORIGIN", default="*"),
any_platform=get_boolean( any_platform=get_boolean(
environ, "ONNX_WEB_ANY_PLATFORM", DEFAULT_ANY_PLATFORM environ, "ONNX_WEB_ANY_PLATFORM", DEFAULT_ANY_PLATFORM
), ),
block_platforms=get_list(environ, "ONNX_WEB_BLOCK_PLATFORMS", ""), block_platforms=get_list(environ, "ONNX_WEB_BLOCK_PLATFORMS"),
default_platform=environ.get("ONNX_WEB_DEFAULT_PLATFORM", None), default_platform=environ.get("ONNX_WEB_DEFAULT_PLATFORM", None),
image_format=environ.get("ONNX_WEB_IMAGE_FORMAT", DEFAULT_IMAGE_FORMAT), image_format=environ.get("ONNX_WEB_IMAGE_FORMAT", DEFAULT_IMAGE_FORMAT),
cache_limit=int(environ.get("ONNX_WEB_CACHE_MODELS", DEFAULT_CACHE_LIMIT)), cache_limit=int(environ.get("ONNX_WEB_CACHE_MODELS", DEFAULT_CACHE_LIMIT)),
show_progress=get_boolean( show_progress=get_boolean(
environ, "ONNX_WEB_SHOW_PROGRESS", DEFAULT_SHOW_PROGRESS environ, "ONNX_WEB_SHOW_PROGRESS", DEFAULT_SHOW_PROGRESS
), ),
optimizations=get_list(environ, "ONNX_WEB_OPTIMIZATIONS", ""), optimizations=get_list(environ, "ONNX_WEB_OPTIMIZATIONS"),
extra_models=get_list(environ, "ONNX_WEB_EXTRA_MODELS", ""), extra_models=get_list(environ, "ONNX_WEB_EXTRA_MODELS"),
job_limit=int(environ.get("ONNX_WEB_JOB_LIMIT", DEFAULT_JOB_LIMIT)), job_limit=int(environ.get("ONNX_WEB_JOB_LIMIT", DEFAULT_JOB_LIMIT)),
memory_limit=memory_limit, memory_limit=memory_limit,
admin_token=environ.get("ONNX_WEB_ADMIN_TOKEN", None), admin_token=environ.get("ONNX_WEB_ADMIN_TOKEN", None),
@ -128,7 +128,7 @@ class ServerContext:
worker_retries=int( worker_retries=int(
environ.get("ONNX_WEB_WORKER_RETRIES", DEFAULT_WORKER_RETRIES) environ.get("ONNX_WEB_WORKER_RETRIES", DEFAULT_WORKER_RETRIES)
), ),
feature_flags=get_list(environ, "ONNX_WEB_FEATURE_FLAGS", ""), feature_flags=get_list(environ, "ONNX_WEB_FEATURE_FLAGS"),
plugins=get_list(environ, "ONNX_WEB_PLUGINS", ""), plugins=get_list(environ, "ONNX_WEB_PLUGINS", ""),
debug=get_boolean(environ, "ONNX_WEB_DEBUG", False), debug=get_boolean(environ, "ONNX_WEB_DEBUG", False),
) )

4
api/onnx_web/utils.py
View File

@ -41,8 +41,8 @@ def get_boolean(args: Any, key: str, default_value: bool) -> bool:
return val.lower() in ("1", "t", "true", "y", "yes") return val.lower() in ("1", "t", "true", "y", "yes")
def get_list(args: Any, key: str) -> List[str]: def get_list(args: Any, key: str, default = "") -> List[str]:
return split_list(args.get(key, "")) return split_list(args.get(key, default))
def get_and_clamp_float( def get_and_clamp_float(