apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

more lint, avoid some unnecessary errors

This commit is contained in:
Sean Sube 2023-12-03 12:57:56 -06:00
parent 9b4ae0916b
commit ae34e466ef
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/onnx_web/chain/result.py
View File

@ -42,7 +42,7 @@ class StageResult:
elif self.images is not None:
return len(self.images)
else:
raise ValueError("empty stage result")
return 0
def as_numpy(self) -> List[np.ndarray]:
if self.arrays is not None:
@ -50,7 +50,7 @@ class StageResult:
elif self.images is not None:
return [np.array(i) for i in self.images]
else:
raise ValueError("empty stage result")
return []
def as_image(self) -> List[Image.Image]:
if self.images is not None:
@ -58,7 +58,7 @@ class StageResult:
elif self.arrays is not None:
return [Image.fromarray(np.uint8(i), shape_mode(i)) for i in self.arrays]
else:
raise ValueError("empty stage result")
return []
def shape_mode(arr: np.ndarray) -> str:

6
api/onnx_web/utils.py
View File

@ -75,13 +75,13 @@ def get_from_list(
def get_from_map(
args: Any, key: str, values: Dict[str, TElem], defaultKey: str
args: Any, key: str, values: Dict[str, TElem], default_key: str
) -> TElem:
selected = args.get(key, defaultKey)
selected = args.get(key, default_key)
if selected in values:
return values[selected]
else:
return values[defaultKey]
return values[default_key]
def get_not_empty(args: Any, key: str, default: TElem) -> TElem: