This commit is contained in:
Auric Vente
2024-07-21 10:47:34 -06:00
parent b1260f894a
commit 9a2c1bdcc5
3 changed files with 15 additions and 1 deletions

View File

@@ -287,8 +287,15 @@ class Game:
@staticmethod
def restart() -> None:
opts = ["25", "50", "100", "250"]
defindex = 0
for i, opt in enumerate(opts):
if int(opt) == Config.default_population:
defindex = i
break
opts = [f"{opt} ants" for opt in opts]
size = Window.prompt_combobox("Size of the population", opts, 2)
size = Window.prompt_combobox("Size of the population", opts, defindex)
if not size:
return