From a4fba209c4d2254a30313f2edc7cb3918ef119f1 Mon Sep 17 00:00:00 2001 From: Jonathan Raiman Date: Wed, 25 Oct 2017 14:08:01 -0700 Subject: [PATCH] remove ref to simple_bench + rank variable not used [mpi gone] --- baselines/acktr/run_atari.py | 2 +- baselines/bench/__init__.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/baselines/acktr/run_atari.py b/baselines/acktr/run_atari.py index 11b9461..ac95180 100644 --- a/baselines/acktr/run_atari.py +++ b/baselines/acktr/run_atari.py @@ -13,7 +13,7 @@ def train(env_id, num_timesteps, seed, num_cpu): def _thunk(): env = make_atari(env_id) env.seed(seed + rank) - env = bench.Monitor(env, logger.get_dir() and os.path.join(logger.get_dir(), str(rank))) + env = bench.Monitor(env, logger.get_dir() and logger.get_dir()) gym.logger.setLevel(logging.WARN) return wrap_deepmind(env) return _thunk diff --git a/baselines/bench/__init__.py b/baselines/bench/__init__.py index 3503929..4cbd5bb 100644 --- a/baselines/bench/__init__.py +++ b/baselines/bench/__init__.py @@ -1,3 +1,2 @@ from baselines.bench.benchmarks import * from baselines.bench.monitor import * -from baselines.bench.simple_bench import simple_bench