From c6a70aae93111046b16dfc7e2a7c07e41a6fcec2 Mon Sep 17 00:00:00 2001
From: Pradnya Khalate <pkhalate@nvidia.com>
Date: Thu, 11 Jan 2024 13:05:23 -0800
Subject: [PATCH] * Update the LD_LIBRARY_PATH and PYTHONPATH for Python wheels
 testing

---
 .github/workflows/test_in_devenv.yml | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/test_in_devenv.yml b/.github/workflows/test_in_devenv.yml
index 22f48d7e5f..12eac52819 100644
--- a/.github/workflows/test_in_devenv.yml
+++ b/.github/workflows/test_in_devenv.yml
@@ -225,8 +225,10 @@ jobs:
             pip install iqm_client --user -vvv
             pip install . --user -vvv
             ## [RPATH-FIX] : Start
-            export LD_LIBRARY_PATH=$CUDAQ_INSTALL_PREFIX/lib${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}
-            export PYTHONPATH=./build${PYTHONPATH:+:$PYTHONPATH}
+            install_loc=$(python3 -m pip show cuda-quantum | grep "Location")
+            install_loc="${loc#*Location: }"
+            export LD_LIBRARY_PATH=$install_loc/lib${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}
+            export PYTHONPATH=./_skbuild${PYTHONPATH:+:$PYTHONPATH}
             ## [RPATH-FIX] : End
             python3 -m pytest -v python/tests/ \
               --ignore python/tests/backends \