python3Packages.jaxlib-build: refactor

NixOS · Jan 20, 2025 · 0906c43 · 0906c43
1 parent 17b7964
commit 0906c43
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix
@@ -6584,10 +6584,9 @@ self: super: with self; {
 
   jaxlib-bin = callPackage ../development/python-modules/jaxlib/bin.nix { };
 
-  jaxlib-build = callPackage ../development/python-modules/jaxlib rec {
-    # Some platforms don't have `cudaSupport` defined, hence the need for 'or false'.
+  jaxlib-build = callPackage ../development/python-modules/jaxlib {
     inherit (pkgs.config) cudaSupport;
-    IOKit = pkgs.darwin.apple_sdk_11_0.IOKit;
+    inherit (pkgs.darwin.apple_sdk_11_0) IOKit;
   };
 
   jaxlib = jaxlib-bin;