* Update the LD_LIBRARY_PATH and PYTHONPATH for Python wheels testing

NVIDIA · Jan 11, 2024 · c6a70aa · c6a70aa
1 parent 860d081
commit c6a70aa
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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 \