diff --git a/pennylane_lightning/lightning_kokkos/lightning_kokkos.py b/pennylane_lightning/lightning_kokkos/lightning_kokkos.py index 42a27e969..b30ca1ad2 100644 --- a/pennylane_lightning/lightning_kokkos/lightning_kokkos.py +++ b/pennylane_lightning/lightning_kokkos/lightning_kokkos.py @@ -534,7 +534,7 @@ def get_c_interface(): # lib.--" # To avoid mismatching the folder name, we search for the shared object instead. # TODO: locate where the naming convention of the folder is decided and replicate it here. - editable_mode_path = package_root.parent.parent / "build_lightning_kokkos" + editable_mode_path = package_root.parent.parent / "build" for path, _, files in os.walk(editable_mode_path): if lib_name in files: lib_location = (Path(path) / lib_name).as_posix() diff --git a/setup.py b/setup.py index 6bd4a6766..aa0502cc7 100644 --- a/setup.py +++ b/setup.py @@ -159,7 +159,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"): source = os.path.join(f"{extdir}", f"lib{backend}_catalyst.so") - destination = os.path.join(os.getcwd(), f"build_{backend}") + destination = os.path.join(os.getcwd(), f"build") shutil.copy(source, destination) with open(os.path.join("pennylane_lightning", "core", "_version.py"), encoding="utf-8") as f: