diff --git a/mattport/structures/colors.py b/mattport/structures/colors.py index 978f44e58..3e039aa95 100644 --- a/mattport/structures/colors.py +++ b/mattport/structures/colors.py @@ -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)}")