columbus/__init__.py

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

columbus/env.py

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

columbus/humanPlayer.py

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