nvidia-container-toolkit: pass nvidia-container-runtime into drv

2 files changed, 8 insertions, 12 deletions

diff --git a/pkgs/applications/virtualization/nvidia-container-toolkit/default.nix b/pkgs/applications/virtualization/nvidia-container-toolkit/default.nix
index fd6edc64a0f6e..1ef9603195d0b 100644
--- a/pkgs/applications/virtualization/nvidia-container-toolkit/default.nix
+++ b/pkgs/applications/virtualization/nvidia-container-toolkit/default.nix
@@ -3,14 +3,8 @@
 , buildGoModule
 , callPackage
 , makeWrapper
-, containerRuntimePath
-, configTemplate
+, nvidia-container-runtime
 }:
-let
-  nvidia-container-runtime = callPackage ../nvidia-container-runtime {
-    inherit containerRuntimePath configTemplate;
-  };
-in
 buildGoModule rec {
   pname = "nvidia-container-toolkit";
   version = "1.3.0";
diff --git a/pkgs/top-level/all-packages.nix b/pkgs/top-level/all-packages.nix
index d90cabd1f0374..ff32fe785933b 100644
--- a/pkgs/top-level/all-packages.nix
+++ b/pkgs/top-level/all-packages.nix
@@ -15416,15 +15416,17 @@ in
   nv-codec-headers = callPackage ../development/libraries/nv-codec-headers { };
 
   mkNvidiaContainerPkg = { name, containerRuntimePath, configTemplate, additionalPaths ? [] }:
-    symlinkJoin {
+    let
+      nvidia-container-runtime = callPackage ../applications/virtualization/nvidia-container-runtime {
+        inherit containerRuntimePath configTemplate;
+      };
+    in symlinkJoin {
       name = "nvidia-containers-${name}";
       paths = [
         (callPackage ../applications/virtualization/libnvidia-container { })
-        (callPackage ../applications/virtualization/nvidia-container-runtime {
-          inherit containerRuntimePath configTemplate;
-        })
+        nvidia-container-runtime
         (callPackage ../applications/virtualization/nvidia-container-toolkit {
-          inherit containerRuntimePath configTemplate;
+          inherit nvidia-container-runtime;
         })
       ] ++ additionalPaths;
     };