pyrox.dev
1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5 packaging,
6 setuptools,
7 torch,
8 kornia-rs,
9}:
10
11buildPythonPackage rec {
12 pname = "kornia";
13 version = "0.8.1";
14 pyproject = true;
15
16 src = fetchFromGitHub {
17 owner = "kornia";
18 repo = "kornia";
19 tag = "v${version}";
20 hash = "sha256-LT+F/tskySvSmaBufIaQhI4+wK5DZBNanQbnYj4ywGo=";
21 };
22
23 build-system = [ setuptools ];
24
25 dependencies = [
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 = {
51 homepage = "https://kornia.readthedocs.io";
52 changelog = "https://github.com/kornia/kornia/releases/tag/${src.tag}";
53 description = "Differentiable computer vision library";
54 license = lib.licenses.asl20;
55 maintainers = with lib.maintainers; [ bcdarwin ];
56 };
57}