add python-mpi4py dependency

2 files changed, 3 insertions, 1 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 130e4079a703..1ecb684806b7 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -9,6 +9,7 @@ pkgbase = python-estool-git
 	depends = python-gym-git
 	depends = python-pybullet
 	depends = python-pycma
+	depends = python-mpi4py
 	optdepends = roboschool: roboschool gym environments
 	provides = python-estool
 	source = git+https://github.com/hardmaru/estool.git
diff --git a/PKGBUILD b/PKGBUILD
index c146a5ae0cc2..58891d5abc3c 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -7,7 +7,8 @@ pkgrel=3
 pkgdesc="Implementation of various Evolution Strategies, such as GA, PEPG, CMA-ES and OpenAI's ES using common interface"
 provides=(python-${_name})
 url="https://github.com/hardmaru/${_name}"
-depends=('python-gym-git' 'python-pybullet' 'python-pycma')
+depends=('python-gym-git' 'python-pybullet' 'python-pycma' 
+'python-mpi4py')
 optdepends=('roboschool: roboschool gym environments')
 makedepends=('python-setuptools')
 license=('MIT')