From ea882cec1d6da86ba40dba6db552aded8b2dbd95 Mon Sep 17 00:00:00 2001
From: alsaiduq-lab <riiidge.racer@gmail.com>
Date: Wed, 21 May 2025 11:32:12 -0600
Subject: [PATCH] updated paths

---
 modules/python.nix | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/modules/python.nix b/modules/python.nix
index d63d450..681e7d2 100644
--- a/modules/python.nix
+++ b/modules/python.nix
@@ -18,6 +18,7 @@
   gccLibPath = "${pkgs.gcc-unwrapped.lib}/lib";
   nvidiaLibPath = "${pkgs.linuxPackages.nvidia_x11}/lib";
   cudaLibPath = "${pkgs.cudatoolkit}/lib";
+  ldLibraryPath = "${gccLibPath}:${nvidiaLibPath}:${cudaLibPath}";
 
   pythonEnv = pkgs.python311.buildEnv.override {
     extraLibs = with py; [
@@ -41,7 +42,7 @@
     extraOutputsToInstall = ["out"];
     postBuild = ''
       wrapProgram $out/bin/python \
-        --prefix LD_LIBRARY_PATH : "${gccLibPath}:${nvidiaLibPath}:${cudaLibPath}"
+        --prefix LD_LIBRARY_PATH : "${ldLibraryPath}"
     '';
   };
 
@@ -51,7 +52,8 @@
     buildInputs = [pkgs.makeWrapper];
     postBuild = ''
       wrapProgram $out/bin/uv \
-        --prefix LD_LIBRARY_PATH : "${gccLibPath}:${nvidiaLibPath}:${cudaLibPath}"
+        --prefix LD_LIBRARY_PATH : "${ldLibraryPath}" \
+        --set PYTHONPATH ""
     '';
   };
 in {
@@ -69,7 +71,7 @@ in {
     ];
     environment.variables = {
       PIP_CONFIG_FILE = "${pipConf}";
-      LD_LIBRARY_PATH = lib.mkForce "${gccLibPath}:${nvidiaLibPath}:${cudaLibPath}";
+      LD_LIBRARY_PATH = lib.mkForce "${ldLibraryPath}";
     };
   };
 }