From 6e7ea8d95b3eb2ca63ed7c557c379167ba058ff2 Mon Sep 17 00:00:00 2001
From: Dave Grote <grote1@llnl.gov>
Date: Wed, 6 Dec 2017 10:58:15 -0800
Subject: Added python callbacks, like beforestep etc

---
 Python/setup.py | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'Python/setup.py')

diff --git a/Python/setup.py b/Python/setup.py
index 20e7c5d9f..71f82dbc7 100644
--- a/Python/setup.py
+++ b/Python/setup.py
@@ -6,9 +6,16 @@ setup.py file for WarpX
 
 from distutils.core import setup
 
+try:
+    from distutils.command.build_py import build_py_2to3 as build_py
+except ImportError:
+    from distutils.command.build_py import build_py
+
+
 setup (name = 'pywarpx',
        packages = ['pywarpx'],
        package_dir = {'pywarpx':'pywarpx'},
        description = """Wrapper of WarpX""",
        package_data = {'pywarpx' : ['libwarpx.so']},
+       cmdclass={'build_py': build_py}
        )
-- 
cgit v1.2.3