From 332e42f59ada4e0e9fd3e5e613d4b512008ebe31 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Mon, 16 Sep 2024 11:16:03 +0200
Subject: [PATCH] Attempt.

---
 .github/workflows/nix_integration_tests.yaml | 2 +-
 flake.nix                                    | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/nix_integration_tests.yaml b/.github/workflows/nix_integration_tests.yaml
index c5cd8e79..5bfec134 100644
--- a/.github/workflows/nix_integration_tests.yaml
+++ b/.github/workflows/nix_integration_tests.yaml
@@ -33,7 +33,7 @@ jobs:
     - name: Python tests.
       run: |
         sudo mkdir -p /run/opengl-driver/lib/
-        export NVIDIA=$(nix-build -E 'with import <nixpkgs> { config = { allowUnfree = true; }; }; linuxPackages.nvidia_x11')
+        export NVIDIA=$(nix build --print-out-paths .#nvidia-driver)
         sudo find $NVIDIA . -name 'libcuda.so*' -exec ln -s {} /run/opengl-driver/lib/ \;
         sudo find $NVIDIA . -name 'libnvidia-ml.so*' -exec ln -s {} /run/opengl-driver/lib/ \;
         nix develop .#test --command pytest -svv integration-tests/ --release
diff --git a/flake.nix b/flake.nix
index 674db0d8..aa6a5597 100644
--- a/flake.nix
+++ b/flake.nix
@@ -149,6 +149,7 @@
           };
         };
 
+        packages.nvidia-driver = pkgs.linuxPackages.nvidia_x11;
         packages.default = pkgs.writeShellApplication {
           name = "text-generation-inference";
           runtimeInputs = [