diff --git a/reasoning_gym/cognition/color_cube_rotation.py b/reasoning_gym/cognition/color_cube_rotation.py index 319780b4..4ec2f699 100644 --- a/reasoning_gym/cognition/color_cube_rotation.py +++ b/reasoning_gym/cognition/color_cube_rotation.py @@ -13,6 +13,16 @@ class Color(StrEnum): YELLOW = "yellow" WHITE = "white" ORANGE = "orange" + PURPLE = "purple" + PINK = "pink" + BROWN = "brown" + GRAY = "gray" + CYAN = "cyan" + MAGENTA = "magenta" + GOLD = "gold" + SILVER = "silver" + INDIGO = "indigo" + VIOLET = "violet" class Side(StrEnum):