diff --git a/setup.py b/setup.py index 0ef706955..018a045cc 100644 --- a/setup.py +++ b/setup.py @@ -157,7 +157,7 @@ def build_extension(self, ext: CMakeExtension): ) # Ensure that catalyst shared object is copied to the build directory for pip editable install - if backend in ("lightning_kokkos"): + if backend in ("lightning_kokkos", "lightning_qubit"): if platform.system() in ["Linux", "Darwin"]: shared_lib_ext = {"Linux": ".so", "Darwin": ".dylib"}[platform.system()] source = os.path.join(f"{extdir}", f"lib{backend}_catalyst{shared_lib_ext}")