update lint

This commit is contained in:
Matt Tancik committed 2022-05-04 15:53:20 -07:00
1 parent fc282f8857
commit 3507467bf4
1 file changed
+3 -3
+3 -3
View File
@@ -32,9 +32,9 @@ def get_color(color: Union[str, list, ListConfig]) -> TensorType[3]:
if color not in COLORS_DICT:
raise ValueError(f"{color} is not a valid preset color")
return COLORS_DICT[color]
elif isinstance(color, list) or isinstance(color, ListConfig):
if isinstance(color, list) or isinstance(color, ListConfig):
if len(color) != 3:
raise ValueError(f"Color should be 3 values (RGB) instead got {color}")
return torch.tensor(color)
else:
raise ValueError(f"Color should be an RGB list or string, instead got {type(color)}")
raise ValueError(f"Color should be an RGB list or string, instead got {type(color)}")