dodox/Columbus
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

smol bug shmashes

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-06-19 16:06:57 +02:00
parent abad4bd6f1
commit 489a0988e7
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
env.py
View File

@ -4,7 +4,8 @@ import numpy as np
import pygame import pygame
import random as random_dont_use import random as random_dont_use
import math import math
from . import entities, observables import entities
import observables
class ColumbusEnv(gym.Env): class ColumbusEnv(gym.Env):
@ -204,4 +205,5 @@ class ColumbusEnv(gym.Env):
class ColumbusTest3_1(ColumbusEnv): class ColumbusTest3_1(ColumbusEnv):
def __init__(self): def __init__(self):
super(ColumbusEnv, self).__init__(observables.CnnObservable()) super(ColumbusTest3_1, self).__init__(
observable=observables.CnnObservable())

4
humanPlayer.py
View File

@ -1,5 +1,5 @@
from time import sleep, time from time import sleep, time
from env import ColumbusEnv from env import ColumbusEnv, ColumbusTest3_1
import numpy as np import numpy as np
import pygame import pygame
@ -7,7 +7,7 @@ from observables import Observable, CnnObservable
def main(): def main():
env = ColumbusEnv(fps=60, observable=CnnObservable()) env = ColumbusTest3_1()
env.start_pos = [0.6, 0.3] env.start_pos = [0.6, 0.3]
playEnv(env) playEnv(env)
env.close() env.close()