smol bug shmashes

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

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