Flake8

Author: eugenevinitsky

examples/train.py

@@ -175,18 +175,16 @@ def on_episode_end(info):
         episode = info["episode"]
         env = info["env"].get_unwrapped()[0]
         avg_speed = np.mean(episode.user_data["avg_speed"])
-        avg_energy = np.mean(episode.user_data["avg_energy"])
+        avg_energy = np.mean(episode.user_data["energy"])
 
         episode.custom_metrics["avg_speed"] = avg_speed
         episode.custom_metrics["avg_energy"] = avg_energy
         episode.custom_metrics["outflow"] = env.k.vehicle.get_outflow_rate()
 
-
     config["callbacks"] = {"on_episode_start": tune.function(on_episode_start),
                            "on_episode_step": tune.function(on_episode_step),
                            "on_episode_end": tune.function(on_episode_end)}
 
-
     # save the flow params for replay
     flow_json = json.dumps(
         flow_params, cls=FlowParamsEncoder, sort_keys=True, indent=4)

flow/utils/rllib.py

@@ -5,7 +5,6 @@
 """
 from copy import deepcopy
 import json
-import sys
 import os
 import sys
 

tests/fast_tests/test_rewards.py

@@ -169,13 +169,13 @@ def test_energy_consumption(self):
 
         # change the speed of one vehicle
         env.k.vehicle.test_set_speed("test_0", 1)
-        self.assertEqual(energy_consumption(env), -12.059337750000001)
+        self.assertEqual(energy_consumption(env), -1.205933775)
 
         # check that stepping change the previous speeds and increases the energy consumption
         env.step(rl_actions=None)
         env.step(rl_actions=None)
         self.assertGreater(env.k.vehicle.get_previous_speed("test_0"), 0.0)
-        self.assertLess(energy_consumption(env), -12.059337750000001)
+        self.assertLess(energy_consumption(env), -1.205933775)
 
     def test_boolean_action_penalty(self):
         """Test the boolean_action_penalty method."""