You could also add a check in Shape to ensure that people pass in valid color keys so that they get the warning on the Python side.