kernels/nix/pkgs/python-modules/kernels-test-utils/default.nix at ef28865ca33a0b8dc81a8cedb5edc71d9fa49b0b · huggingface/kernels · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
{
  lib,
  buildPythonPackage,
  setuptools,

  pytest,
  torch,
}:

let
  version =
    (builtins.fromTOML (builtins.readFile ../../../../kernels-test-utils/pyproject.toml)).project.version;
in
buildPythonPackage {
  pname = "kernels-test-utils";
  inherit version;
  pyproject = true;

  src =
    let
      sourceFiles = file: file.hasExt "toml" || file.hasExt "py";
    in
    lib.fileset.toSource {
      root = ../../../../kernels-test-utils;
      fileset = lib.fileset.fileFilter sourceFiles ../../../../kernels-test-utils;
    };

  build-system = [ setuptools ];

  dependencies = [
    pytest
    torch
  ];

  pythonImportsCheck = [
    "kernels_test_utils"
  ];

  meta = with lib; {
    description = "Shared test utilities for kernel repos";
  };
}