fix benchmarks max_timesteps to be ints

xqding · Jan 31, 2017 · f13b8cc · f13b8cc
1 parent da2dbb9
commit f13b8cc
Showing 1 changed file with 21 additions and 21 deletions.
diff --git a/gym/benchmarks/__init__.py b/gym/benchmarks/__init__.py
@@ -13,49 +13,49 @@
         {
             'env_id': 'BeamRiderNoFrameskip-v3',
             'trials': 2,
-            'max_timesteps': 2e8,
+            'max_timesteps': int(2e8),
             'reward_floor':   363.9,
             'reward_ceiling': 60000.0,
         },
         {
             'env_id': 'BreakoutNoFrameskip-v3',
             'trials': 2,
-            'max_timesteps': 2e8,
+            'max_timesteps': int(2e8),
             'reward_floor':   1.7,
             'reward_ceiling': 800.0,
         },
         {
             'env_id': 'EnduroNoFrameskip-v3',
             'trials': 2,
-            'max_timesteps': 2e8,
+            'max_timesteps': int(2e8),
             'reward_floor':   0.0,
             'reward_ceiling': 5000.0,
         },
         {
             'env_id': 'PongNoFrameskip-v3',
             'trials': 2,
-            'max_timesteps': 2e8,
+            'max_timesteps': int(2e8),
             'reward_floor':  -20.7,
             'reward_ceiling': 21.0,
         },
         {
             'env_id': 'QbertNoFrameskip-v3',
             'trials': 2,
-            'max_timesteps': 2e8,
+            'max_timesteps': int(2e8),
             'reward_floor':   163.9,
             'reward_ceiling': 40000.0,
         },
         {
             'env_id': 'SeaquestNoFrameskip-v3',
             'trials': 2,
-            'max_timesteps': 2e8,
+            'max_timesteps': int(2e8),
             'reward_floor':   68.4,
             'reward_ceiling': 100000.0,
         },
         {
             'env_id': 'SpaceInvadersNoFrameskip-v3',
             'trials': 2,
-            'max_timesteps': 2e8,
+            'max_timesteps': int(2e8),
             'reward_floor':   148.0,
             'reward_ceiling': 30000.0,
         },
@@ -71,49 +71,49 @@
         {
             'env_id': 'BeamRiderNoFrameskip-v3',
             'trials': 2,
-            'max_timesteps': 4e7,
+            'max_timesteps': int(4e7),
             'reward_floor':   363.9,
             'reward_ceiling': 60000.0,
         },
         {
             'env_id': 'BreakoutNoFrameskip-v3',
             'trials': 2,
-            'max_timesteps': 4e7,
+            'max_timesteps': int(4e7),
             'reward_floor':   1.7,
             'reward_ceiling': 800.0,
         },
         {
             'env_id': 'EnduroNoFrameskip-v3',
             'trials': 2,
-            'max_timesteps': 4e7,
+            'max_timesteps': int(4e7),
             'reward_floor':   0.0,
             'reward_ceiling': 5000.0,
         },
         {
             'env_id': 'PongNoFrameskip-v3',
             'trials': 2,
-            'max_timesteps': 4e7,
+            'max_timesteps': int(4e7),
             'reward_floor':  -20.7,
             'reward_ceiling': 21.0,
         },
         {
             'env_id': 'QbertNoFrameskip-v3',
             'trials': 2,
-            'max_timesteps': 4e7,
+            'max_timesteps': int(4e7),
             'reward_floor':   163.9,
             'reward_ceiling': 40000.0,
         },
         {
             'env_id': 'SeaquestNoFrameskip-v3',
             'trials': 2,
-            'max_timesteps': 4e7,
+            'max_timesteps': int(4e7),
             'reward_floor':   68.4,
             'reward_ceiling': 100000.0,
         },
         {
             'env_id': 'SpaceInvadersNoFrameskip-v3',
             'trials': 2,
-            'max_timesteps': 4e7,
+            'max_timesteps': int(4e7),
             'reward_floor':   148.0,
             'reward_ceiling': 30000.0,
         }
@@ -129,49 +129,49 @@
         {
             'env_id': 'FreewayNoFrameskip-v3',
             'trials': 2,
-            'max_timesteps': 4e7,
+            'max_timesteps': int(4e7),
             'reward_floor':   0.1,
             'reward_ceiling': 31.0,
         },
         {
             'env_id': 'GravitarNoFrameskip-v3',
             'trials': 2,
-            'max_timesteps': 4e7,
+            'max_timesteps': int(4e7),
             'reward_floor':   245.5,
             'reward_ceiling': 1000.0,
         },
         {
             'env_id': 'MontezumaRevengeNoFrameskip-v3',
             'trials': 2,
-            'max_timesteps': 4e7,
+            'max_timesteps': int(4e7),
             'reward_floor':   25.0,
             'reward_ceiling': 10000.0,
         },
         {
             'env_id': 'PitfallNoFrameskip-v3',
             'trials': 2,
-            'max_timesteps': 4e7,
+            'max_timesteps': int(4e7),
             'reward_floor':  -348.8,
             'reward_ceiling': 1000.0,
         },
         {
             'env_id': 'PrivateEyeNoFrameskip-v3',
             'trials': 2,
-            'max_timesteps': 4e7,
+            'max_timesteps': int(4e7),
             'reward_floor':   662.8,
             'reward_ceiling': 100.0,
         },
         {
             'env_id': 'SolarisNoFrameskip-v3',
             'trials': 2,
-            'max_timesteps': 4e7,
+            'max_timesteps': int(4e7),
             'reward_floor':   2047.2,
             'reward_ceiling': 5000.0,
         },
         {
             'env_id': 'VentureNoFrameskip-v3',
             'trials': 2,
-            'max_timesteps': 4e7,
+            'max_timesteps': int(4e7),
             'reward_floor':   18.0,
             'reward_ceiling': 100.0,
         }