flake.nix

{
  description = "Flake for yarqa-attn (torch-noarch CPU compartment attention)";

  inputs = {
    kernel-builder.url = "github:huggingface/kernels";
  };

  outputs =
    {
      self,
      kernel-builder,
    }:
    kernel-builder.lib.genKernelFlakeOutputs {
      inherit self;
      path = ./.;
      # pytest is for testshell only. Do not add it as a runtime
      # dependency of the kernel package.
      pythonCheckInputs = pkgs: with pkgs; [ pytest ];
    };
}