From 489a0988e7d373da8751f484b9e387b8750ed81a Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Sun, 19 Jun 2022 16:06:57 +0200
Subject: [PATCH] smol bug shmashes

---
 env.py         | 6 ++++--
 humanPlayer.py | 4 ++--
 2 files changed, 6 insertions(+), 4 deletions(-)

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()