diff --git a/reasoning_gym/algorithmic/graph_color.py b/reasoning_gym/algorithmic/graph_color.py index 54ef39cf..f1775232 100644 --- a/reasoning_gym/algorithmic/graph_color.py +++ b/reasoning_gym/algorithmic/graph_color.py @@ -197,11 +197,11 @@ class GraphColorDataset(ProceduralDataset): question = f""" Please provide a coloring for this graph such that every vertex is not connected to a vertex of the same color. The graph has these properties: -Edges: {edges} Vertices: {puzzle["vertices"]} +Edges: {edges} Possible colors: {puzzle["color_options"]} -Return your solution as a JSON map of verteces to colors. +Return your solution as a JSON map of verteces to colors. (For example: {{0: 1, 1: 2, 2: 3}}) """ return {