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