diff --git a/setup.py b/setup.py index 26ee4410eb6437bbaa2a2117d94bb9dbe12b062e..59151ba6097e534f94a1e97d21ed3542d85b6e4c 100644 --- a/setup.py +++ b/setup.py @@ -83,14 +83,14 @@ class CMakeBuild(build_ext): shutil.copy(currentFile, str(aidge_package.absolute())) # Get "aidge cpu" package - # ext_lib = build_temp / "_CPU" - - # # Copy all shared object files from build_temp/lib to build_lib - # for root, _, files in os.walk(ext_lib.absolute()): - # for file in files: - # if file.endswith('.so'): - # currentFile=os.path.join(root, file) - # shutil.copy(currentFile, str(aidge_package.absolute())) + ext_lib = build_temp / "aidge" / "_CPU" + + # Copy all shared object files from build_temp/lib to build_lib + for root, _, files in os.walk(ext_lib.absolute()): + for file in files: + if file.endswith('.so'): + currentFile=os.path.join(root, file) + shutil.copy(currentFile, str(aidge_package.absolute())) if __name__ == '__main__':