apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix default params

This commit is contained in:
Sean Sube 2024-02-24 11:23:57 -06:00
parent 500423424e
commit 09452376f5
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/onnx_web/utils.py
View File

@ -50,21 +50,21 @@ def get_boolean(args: Any, key: str, default_value: bool) -> bool:
def get_list(args: Any, key: str, default="") -> List[str]: def get_list(args: Any, key: str, default="") -> List[str]:
val = recursive_get(args, key.split("."), default=default) val = recursive_get(args, key.split("."), default_value=default)
return split_list(val) return split_list(val)
def get_and_clamp_float( def get_and_clamp_float(
args: Any, key: str, default_value: float, max_value: float, min_value=0.0 args: Any, key: str, default_value: float, max_value: float, min_value=0.0
) -> float: ) -> float:
val = recursive_get(args, key.split("."), default=default_value) val = recursive_get(args, key.split("."), default_value=default_value)
return min(max(float(val), min_value), max_value) return min(max(float(val), min_value), max_value)
def get_and_clamp_int( def get_and_clamp_int(
args: Any, key: str, default_value: int, max_value: int, min_value=1 args: Any, key: str, default_value: int, max_value: int, min_value=1
) -> int: ) -> int:
val = recursive_get(args, key.split("."), default=default_value) val = recursive_get(args, key.split("."), default_value=default_value)
return min(max(int(val), min_value), max_value) return min(max(int(val), min_value), max_value)