From 309d93ef9f47275ce9dccbb0b6671ab6fee267fb Mon Sep 17 00:00:00 2001 From: matixezor Date: Tue, 16 Mar 2021 20:22:27 +0100 Subject: [PATCH] code optimization after piotrek fucked up --- src/environment.py | 2 +- src/main.py | 5 ++--- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/src/environment.py b/src/environment.py index 9d38dc8..21e1725 100644 --- a/src/environment.py +++ b/src/environment.py @@ -3,7 +3,7 @@ from random import choices def generate_field() -> List[List[int]]: - return [choices(range(9), weights=[8, 8, 4, 4, 4, 4,4,4,4], k=10) for _ in range(10)] + return [choices(range(9), weights=[8, 8, 4, 4, 4, 4, 4, 4, 4], k=10) for _ in range(10)] class Environment: diff --git a/src/main.py b/src/main.py index df467be..198f84f 100644 --- a/src/main.py +++ b/src/main.py @@ -3,8 +3,7 @@ import pygame as pg from const import ICON from agent import Agent from game_ui import GameUi -from const import DEFAULT_FIELD -from environment import Environment,generate_field +from environment import Environment def main(): @@ -12,7 +11,7 @@ def main(): pg.display.set_caption('Super Saper') pg.display.set_icon(pg.image.load(ICON)) - env = Environment(generate_field()) + env = Environment() agent = Agent() game_ui = GameUi(agent, env) game_ui.update()