Make test shells work again

flake.lock

@@ -73,11 +73,11 @@
         "nixpkgs": "nixpkgs"
       },
       "locked": {
-        "lastModified": 1757493151,
-        "narHash": "sha256-eirWlcvs2rjZmU8JcF4CKN1IEnNfpQnGuf2qbK3IQh8=",
+        "lastModified": 1757675377,
+        "narHash": "sha256-JQKZOI1ZYO4faJnanuoTXziSmqzXe5rEFSGliWDWqWw=",
         "owner": "huggingface",
         "repo": "hf-nix",
-        "rev": "503cd4eb9866103c983dbef93d9ad5db4fb6b415",
+        "rev": "faf3354403a7381958d08e826c15fe30f6986a4f",
         "type": "github"
       },
       "original": {
@@ -98,15 +98,16 @@
         ]
       },
       "locked": {
-        "lastModified": 1757570810,
-        "narHash": "sha256-YFWQwy2LKbhjdLW8wkyNkE/+Vbdn6qlJif2CKvBT9Qo=",
+        "lastModified": 1757932094,
+        "narHash": "sha256-BRhrwYAW25v1SsEmAEqSXwUSHG4KnLw0Ttc5K9IGPOk=",
         "owner": "huggingface",
         "repo": "kernel-builder",
-        "rev": "1201847af3ff757b65015c6e06b5bd75896d2d4b",
+        "rev": "4695b186164ac7d81a86cb4e27f2a3a0e1b5b11a",
         "type": "github"
       },
       "original": {
         "owner": "huggingface",
+        "ref": "fix-cuda-test-shells",
         "repo": "kernel-builder",
         "type": "github"
       }

flake.nix

@@ -1,7 +1,9 @@
 {
   description = "Flake for Torch kernel extension";
   inputs = {
-    kernel-builder.url = "github:huggingface/kernel-builder";
+    # Go back to main after https://github.com/huggingface/kernel-builder/pull/217
+    # is merged.
+    kernel-builder.url = "github:huggingface/kernel-builder/fix-cuda-test-shells";
   };
   outputs = { self, kernel-builder, }:
     kernel-builder.lib.genFlakeOutputs {