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