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