Skip to content

Commit

Permalink
Flake8
Browse files Browse the repository at this point in the history
  • Loading branch information
eugenevinitsky committed Mar 19, 2020
1 parent 2183ca5 commit 243979f
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 3 deletions.
2 changes: 0 additions & 2 deletions examples/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down
1 change: 0 additions & 1 deletion flow/utils/rllib.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,6 @@
"""
from copy import deepcopy
import json
import sys
import os
import sys

Expand Down

0 comments on commit 243979f

Please sign in to comment.