The 2d and 3d Python versions can both be installed

1 files changed, 2 insertions, 2 deletions

diff --git a/Python/setup.py b/Python/setup.py
index 5f762dabd..e68870809 100644
--- a/Python/setup.py
+++ b/Python/setup.py
@@ -15,12 +15,12 @@ except ImportError:
     from distutils.command.build_py import build_py
 
 argparser = argparse.ArgumentParser(add_help=False)
-argparser.add_argument('--with-libwarpx', action='store_true', help='Install libwarpx. This option is only used by the makefile.')
+argparser.add_argument('--with-libwarpx', type=int, default=None, help='Install libwarpx with the given value as DIM. This option is only used by the makefile.')
 args, unknown = argparser.parse_known_args()
 sys.argv = [sys.argv[0]] + unknown
 
 if args.with_libwarpx:
-    package_data = {'pywarpx' : ['libwarpx.so']}
+    package_data = {'pywarpx' : ['libwarpx%dd.so'%args.with_libwarpx]}
 else:
     package_data = {}