From 4010fafd1d7b36c9aaa1fcb1bcc67be5454ce13a Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Thu, 3 Aug 2023 14:36:20 +0000
Subject: [PATCH] Working setup.py.

---
 .gitignore             | 1 +
 aidge_core/__init__.py | 2 +-
 setup.py               | 4 ++--
 3 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/.gitignore b/.gitignore
index 9fbfccca6..f37378e30 100644
--- a/.gitignore
+++ b/.gitignore
@@ -10,6 +10,7 @@ install*/
 __pycache__
 *.pyc
 *.egg-info
+dist*/
 
 # Mermaid
 *.mmd
diff --git a/aidge_core/__init__.py b/aidge_core/__init__.py
index 3b9dbbf8f..8e0bef3a0 100644
--- a/aidge_core/__init__.py
+++ b/aidge_core/__init__.py
@@ -1 +1 @@
-from .aidge_core import * # import so generated by PyBind
\ No newline at end of file
+from aidge_core.aidge_core import * # import so generated by PyBind
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 7dbde6a0a..cc3d5dbb8 100644
--- a/setup.py
+++ b/setup.py
@@ -102,10 +102,10 @@ if __name__ == '__main__':
         platforms=["Linux"],
         packages=find_packages(where="."),
         include_package_data=True,
-        # exclude_package_data={"aidge": ["_Core/*", "_CPU/*"]},
-        ext_modules=[CMakeExtension('aidge_core')],
+        ext_modules=[CMakeExtension(get_project_name())],
         cmdclass={
             'build_ext': CMakeBuild,
         },
+        zip_safe=False,
 
     )
-- 
GitLab