From c6c5bfc349cb557db47aa0f92153caef5c0e3f15 Mon Sep 17 00:00:00 2001
From: Lorenzo Giacomel <lorenzo.giacomel@cern.ch>
Date: Tue, 30 Aug 2022 11:20:15 +0200
Subject: [PATCH] Added mpi4py as a dependency if --mpi-parallel is specified

---
 setup.py | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/setup.py b/setup.py
index d1ee80af..ec903869 100755
--- a/setup.py
+++ b/setup.py
@@ -65,6 +65,14 @@
               library_dirs=[], libraries=["m"])
 ]
 
+
+flag_mpi = "--mpi-parallel" in sys.argv
+setup_reqires ['h5py', 'numpy', 'scipy', 'cython']
+
+if "--mpi-parallel" in sys.argv:
+    setup_requires += ['mpi4py']
+
+
 setup(
     name='PyHEADTAIL',
     version=__version__,
@@ -80,12 +88,7 @@
     cmdclass={'build_ext': build_ext},
     ext_modules=cythonize(cy_ext, **cy_ext_options),
     include_package_data=True,  # install files matched by MANIFEST.in
-    setup_requires=[
-        'h5py',
-        'numpy',
-        'scipy',
-        'cython',
-    ]
+    setup_requires=setup_requires
 )
 
 # from numpy.distutils.core import setup, Extension