From 3637f218ddab6f4db78b2cbff68ef1241d50633f Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Wed, 26 Mar 2025 18:30:19 +0100
Subject: [PATCH] Fixing the impure flake devShell to be able to run python
 code.

---
 flake.nix | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/flake.nix b/flake.nix
index ff3ae6d0..1477e238 100644
--- a/flake.nix
+++ b/flake.nix
@@ -206,8 +206,11 @@
                 openssl
                 pkg-config
                 cudaPackages.cudatoolkit
+                python3Packages.python
+                python3Packages.venvShellHook
               ];
-              LD_LIBRARY_PATH = "/run/opengl-driver/lib";
+              venvDir = "./.venv";
+              LD_LIBRARY_PATH = "${pkgs.stdenv.cc.cc.lib}/lib:/run/opengl-driver/lib";
               LIBRARY_PATH = "${pkgs.stdenv.cc.cc.lib}/lib:/run/opengl-driver/lib";
               CUDA_ROOT = "${pkgs.cudaPackages.cudatoolkit}";
             };