1 files changed, 8 insertions, 3 deletions
diff --git a/modules/nixos/traits/containers.nix b/modules/nixos/traits/containers.nix
index 1399dbf..407ee51 100644
--- a/modules/nixos/traits/containers.nix
+++ b/modules/nixos/traits/containers.nix
@@ -6,11 +6,16 @@
 }:
 let
   cfg = config.traits.containers;
-  enableNvidia = lib.elem "nvidia" (config.services.xserver.videoDrivers or [ ]);
 in
 {
   options.traits.containers = {
-    enable = lib.mkEnableOption "containers support";
+    enable = lib.mkEnableOption "support for containers";
+    enableNvidia = lib.mkEnableOption "the use of nvidia-container-toolkit" // {
+      default = lib.elem "nvidia" config.services.xserver.videoDrivers;
+      defaultText = lib.literalExpression ''
+        lib.elem "nvidia" config.services.xserver.videoDrivers
+      '';
+    };
   };
 
   config = lib.mkIf cfg.enable (
@@ -27,7 +32,7 @@ in
         };
       }
 
-      (lib.mkIf enableNvidia { hardware.nvidia-container-toolkit.enable = true; })
+      (lib.mkIf cfg.enableNvidia { hardware.nvidia-container-toolkit.enable = true; })
     ]
   );
 }