Enhance extension building in setup

Updated the `BuildExtCommand` class in `setup.py` to append `rpath`
flags and link against `hdf5` libraries when building extensions. This
ensures that the shared libraries are correctly located at runtime and
resolves potential linkage issues with HDF5-related extensions for a
smoother installation and deployment process.
This commit is contained in:
Kumi 2024-02-09 15:33:51 +01:00
parent 75b7340354
commit c0d1fbbf66
Signed by: kumi
GPG key ID: ECBCC9082395383F

View file

@ -12,6 +12,13 @@ class BuildExtCommand(build_ext):
"""Custom build_ext command to ensure that the submodule is retrieved and built."""
def build_extensions(self):
rpath_flags = ['-Wl,-rpath,/usr/lib/']
for ext in self.extensions:
ext.extra_compile_args = rpath_flags
ext.extra_link_args = rpath_flags
ext.libraries.extend(['hdf5', 'hdf5_cpp'])
build_ext.build_extensions(self)
super().build_extensions()
def run(self):