further removing MPI references where unnecessary

This commit is contained in:
Peter Zhokhov
2018-10-29 15:38:16 -07:00
parent 064c45fa76
commit ea7a52b652

View File

@@ -10,9 +10,9 @@ from baselines.common import explained_variance, set_global_seeds
from baselines.common.policies import build_policy from baselines.common.policies import build_policy
from baselines.common.runners import AbstractEnvRunner from baselines.common.runners import AbstractEnvRunner
from baselines.common.tf_util import get_session, save_variables, load_variables from baselines.common.tf_util import get_session, save_variables, load_variables
from baselines.common.mpi_adam_optimizer import MpiAdamOptimizer
try: try:
from baselines.common.mpi_adam_optimizer import MpiAdamOptimizer
from mpi4py import MPI from mpi4py import MPI
except ImportError: except ImportError:
MPI = None MPI = None