From ea7a52b652adb86d3c6962480550b0a300503cda Mon Sep 17 00:00:00 2001 From: Peter Zhokhov Date: Mon, 29 Oct 2018 15:38:16 -0700 Subject: [PATCH] further removing MPI references where unnecessary --- baselines/ppo2/ppo2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/baselines/ppo2/ppo2.py b/baselines/ppo2/ppo2.py index fdd52d0..9a5896f 100644 --- a/baselines/ppo2/ppo2.py +++ b/baselines/ppo2/ppo2.py @@ -10,9 +10,9 @@ from baselines.common import explained_variance, set_global_seeds from baselines.common.policies import build_policy from baselines.common.runners import AbstractEnvRunner from baselines.common.tf_util import get_session, save_variables, load_variables -from baselines.common.mpi_adam_optimizer import MpiAdamOptimizer try: + from baselines.common.mpi_adam_optimizer import MpiAdamOptimizer from mpi4py import MPI except ImportError: MPI = None