make baselines run without mpi wip

2018-10-19 17:00:41 -07:00
parent d0cc325e14
commit d96e20ff27
6 changed files with 56 additions and 29 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'