mpi-less baselines (#689)

* make baselines run without mpi wip * squash-merged latest master * further removing MPI references where unnecessary * more MPI removal * syntax and flake8 * MpiAdam becomes regular Adam if Mpi not present * autopep8 * add assertion to test in mpi_adam; fix trpo_mpi failure without MPI on cartpole * mpiless ddpg
2018-10-31 11:15:41 -07:00
parent a071fa7630
commit ab59de6922
8 changed files with 124 additions and 45 deletions
--- a/setup.py
+++ b/setup.py
@@ -15,6 +15,9 @@ extras = {
    ],
    'bullet': [
        'pybullet',
+    ],
+    'mpi': [
+        'mpi4py'
    ]
 }

@@ -34,7 +37,6 @@ setup(name='baselines',
          'joblib',
          'dill',
          'progressbar2',
-          'mpi4py',
          'cloudpickle',
          'click',
          'opencv-python'