python3Packages.kernels-data: init

Author: danieldk

flake.nix

@@ -166,6 +166,7 @@
               jax
               jax-tvm-ffi
               kernel-abi-check
+              kernels-data
               matplotlib
               mktestdocs
               openssl.dev

nix-builder/overlay.nix

@@ -158,6 +158,8 @@ in
 
         kernels = callPackage ./pkgs/python-modules/kernels { };
 
+        kernels-data = callPackage ./pkgs/python-modules/kernels-data { };
+
         pyclibrary = python-self.callPackage ./pkgs/python-modules/pyclibrary { };
 
         mkTorch = callPackage ./pkgs/python-modules/torch/binary { };

nix-builder/pkgs/crate-dirs.nix

@@ -0,0 +1,13 @@
+{ lib, sourceFiles }:
+
+with lib.fileset;
+toSource {
+  root = ../..;
+  fileset = unions [
+    ../../Cargo.lock
+    ../../Cargo.toml
+    (fileFilter sourceFiles ../../kernel-abi-check)
+    (fileFilter sourceFiles ../../kernel-builder)
+    (fileFilter sourceFiles ../../kernels-data)
+  ];
+}

nix-builder/pkgs/kernel-abi-check/default.nix

@@ -26,17 +26,8 @@ rustPlatform.buildRustPackage {
         || file.hasExt "rs"
         || file.name == "stable_abi.toml";
     in
-    with lib.fileset;
-    toSource {
-      root = ../../..;
-      fileset = unions [
-        ../../../Cargo.lock
-        ../../../Cargo.toml
-        (fileFilter sourceFiles ../../../kernel-abi-check)
-        # Cargo wants access to the whole workspace.
-        (fileFilter sourceFiles ../../../kernel-builder)
-        (fileFilter sourceFiles ../../../kernels-data)
-      ];
+    import ../crate-dirs.nix {
+      inherit lib sourceFiles;
     };
 
   cargoLock = {

nix-builder/pkgs/kernel-builder/default.nix

@@ -45,17 +45,8 @@ rustPlatform.buildRustPackage {
           "toml"
         ]);
     in
-    with lib.fileset;
-    toSource {
-      root = ../../..;
-      fileset = unions [
-        ../../../Cargo.lock
-        ../../../Cargo.toml
-        (fileFilter sourceFiles ../../../kernel-builder)
-        # Cargo wants access to the whole workspace.
-        (fileFilter sourceFiles ../../../kernel-abi-check)
-        (fileFilter sourceFiles ../../../kernels-data)
-      ];
+    import ../crate-dirs.nix {
+      inherit lib sourceFiles;
     };
 
   cargoLock = {

nix-builder/pkgs/python-modules/kernel-abi-check/default.nix

@@ -30,17 +30,8 @@ buildPythonPackage {
         || file.hasExt "rs"
         || file.name == "stable_abi.toml";
     in
-    with lib.fileset;
-    toSource {
-      root = ../../../..;
-      fileset = unions [
-        ../../../../Cargo.lock
-        ../../../../Cargo.toml
-        (fileFilter sourceFiles ../../../../kernel-abi-check)
-        # Cargo wants access to the whole workspace.
-        (fileFilter sourceFiles ../../../../kernel-builder)
-        (fileFilter sourceFiles ../../../../kernels-data)
-      ];
+    import ../../crate-dirs.nix {
+      inherit lib sourceFiles;
     };
 
   cargoDeps = rustPlatform.importCargoLock {

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

@@ -0,0 +1,52 @@
+{
+  lib,
+  buildPythonPackage,
+  rustPlatform,
+}:
+
+let
+  version =
+    (builtins.fromTOML (builtins.readFile ../../../../kernels-data/Cargo.toml)).package.version;
+  cargoFlags = [
+    "-m"
+    "kernels-data/bindings/python/Cargo.toml"
+  ];
+in
+buildPythonPackage {
+  pname = "kernels-data";
+  inherit version;
+  format = "pyproject";
+
+  src =
+    let
+      sourceFiles =
+        file:
+        file.name == "Cargo.toml"
+        || file.name == "Cargo.lock"
+        || file.hasExt "pyi"
+        || file.name == "pyproject.toml"
+        || file.name == "python_dependencies.json"
+        || file.hasExt "rs";
+    in
+    import ../../crate-dirs.nix {
+      inherit lib sourceFiles;
+    };
+
+  cargoDeps = rustPlatform.importCargoLock {
+    lockFile = ../../../../Cargo.lock;
+    outputHashes = {
+      "huggingface-hub-0.0.1" = "sha256-By8b1NUPWu+XF3Om1NcEO+o2qdZUco+FxvrJGNRqxWs=";
+    };
+  };
+
+  maturinBuildFlags = cargoFlags;
+
+  build-system = [
+    rustPlatform.cargoSetupHook
+    rustPlatform.maturinBuildHook
+  ];
+
+  meta = with lib; {
+    description = "Kernels data structures";
+  };
+}