Gave all envs a max time of 2min

columbus/__init__.py

@@ -1,2 +1 @@
-# emtpy
 from columbus import env

columbus/env.py

@@ -386,28 +386,29 @@ class ColumbusStateWithBarriers(ColumbusEnv):
 
 
 ###
+
 register(
     id='ColumbusTestCnn-v0',
     entry_point=ColumbusTest3_1,
-    max_episode_steps=30*60*5,
+    max_episode_steps=30*60*2,
 )
 
 register(
     id='ColumbusTestRay-v0',
     entry_point=ColumbusTestRay,
-    max_episode_steps=30*60*5,
+    max_episode_steps=30*60*2,
 )
 
 register(
     id='ColumbusRayDrone-v0',
     entry_point=ColumbusRayDrone,
-    max_episode_steps=30*60*5,
+    max_episode_steps=30*60*2,
 )
 
 register(
     id='ColumbusCandyland-v0',
     entry_point=ColumbusCandyland,
-    max_episode_steps=30*60*5,
+    max_episode_steps=30*60*2,
 )
 
 register(

columbus/humanPlayer.py

@@ -1,9 +1,9 @@
 from time import sleep, time
-import env
 import numpy as np
 import pygame
 
-from observables import Observable, CnnObservable
+from columbus import env
+from columbus.observables import Observable, CnnObservable
 
 
 def main():