Flake8

Author: eugenevinitsky

examples/train.py

@@ -181,12 +181,10 @@ def on_episode_end(info):
         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