diff --git a/reasoning_gym/graphs/largest_island.py b/reasoning_gym/graphs/largest_island.py index 364d2687..ee2615e7 100644 --- a/reasoning_gym/graphs/largest_island.py +++ b/reasoning_gym/graphs/largest_island.py @@ -31,8 +31,12 @@ class LargestIslandConfig: rows: int = 10 # Number of rows in the grid cols: int = 10 # Number of columns in the grid - max_num_islands: int = 5 # Maximum number of islands (actual max might be smaller due to merging of islands during random walk) - max_island_size: int = 10 # Maximum size of an island (actual max might be larger due to merging of islands during random walk) + max_num_islands: int = ( + 5 # Maximum number of islands (actual max might be smaller due to merging of islands during random walk) + ) + max_island_size: int = ( + 10 # Maximum size of an island (actual max might be larger due to merging of islands during random walk) + ) size: int = 500 # Virtual dataset size seed: Optional[int] = None