bindings: Example of using pybind11 with numpy

1 files changed, 22 insertions, 0 deletions

diff --git a/bindings/setup.py b/bindings/setup.py
new file mode 100644
index 0000000..487949a
--- /dev/null
+++ b/bindings/setup.py
@@ -0,0 +1,22 @@
+from setuptools import setup, Extension
+import pybind11
+import numpy
+
+ext_modules = [
+    Extension(
+        'galearn_pdm',
+        ['galearn_pdm.cpp'],
+        include_dirs=[
+            pybind11.get_include(),
+            numpy.get_include(),
+        ],
+        language='c++',
+    ),
+]
+
+setup(
+    name='galearn',
+    version='0.0.1',
+    ext_modules=ext_modules,
+    zip_safe=False,
+)