tjh.dev
1{ lib
2, buildPythonPackage
3, fetchFromGitHub
4, pythonOlder
5, pytestCheckHook
6, packaging
7, torch
8}:
9
10buildPythonPackage rec {
11 pname = "kornia";
12 version = "0.6.12";
13 format = "pyproject";
14
15 disabled = pythonOlder "3.7";
16
17 src = fetchFromGitHub {
18 owner = pname;
19 repo = pname;
20 rev = "refs/tags/v${version}";
21 hash = "sha256-qLJos1ivEws/jFK4j0Kp1ij9J9ZwCoHFRYXnlYxwPFY=";
22 };
23
24 propagatedBuildInputs = [
25 packaging
26 torch
27 ];
28
29 pythonImportsCheck = [
30 "kornia"
31 "kornia.augmentation"
32 "kornia.color"
33 "kornia.contrib"
34 "kornia.enhance"
35 "kornia.feature"
36 "kornia.filters"
37 "kornia.geometry"
38 "kornia.io"
39 "kornia.losses"
40 "kornia.metrics"
41 "kornia.morphology"
42 "kornia.tracking"
43 "kornia.testing"
44 "kornia.utils"
45 ];
46
47 doCheck = false; # tests hang with no single test clearly responsible
48
49 meta = with lib; {
50 homepage = "https://kornia.github.io/kornia";
51 changelog = "https://github.com/kornia/kornia/releases/tag/v${version}";
52 description = "Differentiable computer vision library";
53 license = licenses.asl20;
54 maintainers = with maintainers; [ bcdarwin ];
55 };
56}