From 5c5a9f4b31fed03f33f512d48d497ddde1450d36 Mon Sep 17 00:00:00 2001 From: Peter Zhokhov Date: Tue, 11 Sep 2018 12:47:50 -0700 Subject: [PATCH] autopep8 on deepq/experiments --- baselines/deepq/experiments/enjoy_mountaincar.py | 2 +- baselines/deepq/experiments/train_pong.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/baselines/deepq/experiments/enjoy_mountaincar.py b/baselines/deepq/experiments/enjoy_mountaincar.py index 8b1089e..2998bb6 100644 --- a/baselines/deepq/experiments/enjoy_mountaincar.py +++ b/baselines/deepq/experiments/enjoy_mountaincar.py @@ -7,7 +7,7 @@ from baselines.common import models def main(): env = gym.make("MountainCar-v0") act = deepq.learn( - env, + env, network=models.mlp(num_layers=1, num_hidden=64), total_timesteps=0, load_path='mountaincar_model.pkl' diff --git a/baselines/deepq/experiments/train_pong.py b/baselines/deepq/experiments/train_pong.py index a8febb9..b031021 100644 --- a/baselines/deepq/experiments/train_pong.py +++ b/baselines/deepq/experiments/train_pong.py @@ -29,7 +29,6 @@ def main(): model.save('pong_model.pkl') env.close() - if __name__ == '__main__':