more lint, avoid some unnecessary errors
This commit is contained in:
parent
9b4ae0916b
commit
ae34e466ef
|
@ -42,7 +42,7 @@ class StageResult:
|
||||||
elif self.images is not None:
|
elif self.images is not None:
|
||||||
return len(self.images)
|
return len(self.images)
|
||||||
else:
|
else:
|
||||||
raise ValueError("empty stage result")
|
return 0
|
||||||
|
|
||||||
def as_numpy(self) -> List[np.ndarray]:
|
def as_numpy(self) -> List[np.ndarray]:
|
||||||
if self.arrays is not None:
|
if self.arrays is not None:
|
||||||
|
@ -50,7 +50,7 @@ class StageResult:
|
||||||
elif self.images is not None:
|
elif self.images is not None:
|
||||||
return [np.array(i) for i in self.images]
|
return [np.array(i) for i in self.images]
|
||||||
else:
|
else:
|
||||||
raise ValueError("empty stage result")
|
return []
|
||||||
|
|
||||||
def as_image(self) -> List[Image.Image]:
|
def as_image(self) -> List[Image.Image]:
|
||||||
if self.images is not None:
|
if self.images is not None:
|
||||||
|
@ -58,7 +58,7 @@ class StageResult:
|
||||||
elif self.arrays is not None:
|
elif self.arrays is not None:
|
||||||
return [Image.fromarray(np.uint8(i), shape_mode(i)) for i in self.arrays]
|
return [Image.fromarray(np.uint8(i), shape_mode(i)) for i in self.arrays]
|
||||||
else:
|
else:
|
||||||
raise ValueError("empty stage result")
|
return []
|
||||||
|
|
||||||
|
|
||||||
def shape_mode(arr: np.ndarray) -> str:
|
def shape_mode(arr: np.ndarray) -> str:
|
||||||
|
|
|
@ -75,13 +75,13 @@ def get_from_list(
|
||||||
|
|
||||||
|
|
||||||
def get_from_map(
|
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:
|
) -> TElem:
|
||||||
selected = args.get(key, defaultKey)
|
selected = args.get(key, default_key)
|
||||||
if selected in values:
|
if selected in values:
|
||||||
return values[selected]
|
return values[selected]
|
||||||
else:
|
else:
|
||||||
return values[defaultKey]
|
return values[default_key]
|
||||||
|
|
||||||
|
|
||||||
def get_not_empty(args: Any, key: str, default: TElem) -> TElem:
|
def get_not_empty(args: Any, key: str, default: TElem) -> TElem:
|
||||||
|
|
Loading…
Reference in New Issue