diff --git a/env.py b/env.py index e8aec68..5cf25c0 100644 --- a/env.py +++ b/env.py @@ -4,7 +4,8 @@ import numpy as np import pygame import random as random_dont_use import math -from . import entities, observables +import entities +import observables class ColumbusEnv(gym.Env): @@ -204,4 +205,5 @@ class ColumbusEnv(gym.Env): class ColumbusTest3_1(ColumbusEnv): def __init__(self): - super(ColumbusEnv, self).__init__(observables.CnnObservable()) + super(ColumbusTest3_1, self).__init__( + observable=observables.CnnObservable()) diff --git a/humanPlayer.py b/humanPlayer.py index bb44b6b..ad3f999 100644 --- a/humanPlayer.py +++ b/humanPlayer.py @@ -1,5 +1,5 @@ from time import sleep, time -from env import ColumbusEnv +from env import ColumbusEnv, ColumbusTest3_1 import numpy as np import pygame @@ -7,7 +7,7 @@ from observables import Observable, CnnObservable def main(): - env = ColumbusEnv(fps=60, observable=CnnObservable()) + env = ColumbusTest3_1() env.start_pos = [0.6, 0.3] playEnv(env) env.close()