python3Packages.kernels: build from local source (#227)

Author: danieldk

flake.nix

@@ -190,6 +190,7 @@
 
         packages = rec {
           inherit (buildSet.pkgs) build2cmake kernel-abi-check;
+          inherit (buildSet.pkgs.python3.pkgs) kernels;
 
           update-build = pkgs.writeShellScriptBin "update-build" ''
             ${build2cmake}/bin/build2cmake update-build ''${1:-build.toml}
@@ -210,7 +211,6 @@
               value = buildSet.torch;
             }) buildSets
           );
-
         };
       }
     )

nix/overlay.nix

@@ -81,16 +81,7 @@ in
 
         kernel-abi-check = callPackage ./pkgs/python-modules/kernel-abi-check { };
 
-        kernels = python-super.kernels.overrideAttrs (oldAttrs: rec {
-          version = "0.11.6";
-
-          src = final.fetchFromGitHub {
-            owner = "huggingface";
-            repo = "kernels";
-            rev = "ff4de4eafdfe5c2346b252cf09c6c412e13f3135";
-            hash = "sha256-hn9PZsGnd2BsBejjDQXVKye8abnS5K9jWga7YVkvXOg=";
-          };
-        });
+        kernels = callPackage ./pkgs/python-modules/kernels { };
 
         pyclibrary = python-self.callPackage ./pkgs/python-modules/pyclibrary { };
 

nix/pkgs/python-modules/kernels/default.nix

@@ -0,0 +1,47 @@
+{
+  lib,
+  buildPythonPackage,
+  setuptools,
+
+  huggingface-hub,
+  kernel-abi-check,
+  pyyaml,
+  torch,
+}:
+
+let
+  version =
+    (builtins.fromTOML (builtins.readFile ../../../../kernels/pyproject.toml)).project.version;
+in
+buildPythonPackage {
+  pname = "kernels";
+  inherit version;
+  pyproject = true;
+
+  src =
+    let
+      sourceFiles =
+        file: file.hasExt "lock" || file.hasExt "json" || file.hasExt "toml" || file.hasExt "py";
+    in
+    lib.fileset.toSource {
+      root = ../../../../kernels;
+      fileset = lib.fileset.fileFilter sourceFiles ../../../../kernels;
+    };
+
+  build-system = [ setuptools ];
+
+  dependencies = [
+    huggingface-hub
+    kernel-abi-check
+    pyyaml
+    torch
+  ];
+
+  pythonImportsCheck = [
+    "kernels"
+  ];
+
+  meta = with lib; {
+    description = "Python client for the Kernel Hub";
+  };
+}