tjh.dev
1{ buildPythonPackage
2, fetchFromGitHub
3, pythonOlder
4, pytestCheckHook
5, dill
6, h5py
7, hickle
8, numpy
9, opencv4
10, pandas
11, pillow
12, pycm
13, pyyaml
14, scipy
15, requests
16, scikit-image
17, six
18, tabulate
19, torch
20, torchvision
21, tqdm
22, yacs
23, lib
24}:
25
26buildPythonPackage rec {
27 pname = "pywick";
28 version = "0.6.5";
29
30 disabled = pythonOlder "3.6";
31
32 src = fetchFromGitHub {
33 owner = "achaiah";
34 repo = pname;
35 rev = "v${version}";
36 sha256 = "0wnijdvqgdpzfdsy1cga3bsr0n7zzsl8hp4dskqwxx087g5h1r84";
37 };
38
39 postPatch = ''
40 substituteInPlace setup.py --replace "opencv-python-headless" "opencv"
41 '';
42
43 propagatedBuildInputs = [
44 dill
45 h5py
46 hickle
47 numpy
48 opencv4
49 pandas
50 pillow
51 pycm
52 pyyaml
53 scipy
54 requests
55 scikit-image
56 tabulate
57 torch
58 torchvision
59 tqdm
60 six
61 yacs
62 ];
63
64 nativeCheckInputs = [ pytestCheckHook ];
65
66 meta = {
67 description = "High-level training framework for Pytorch";
68 homepage = "https://github.com/achaiah/pywick";
69 license = lib.licenses.mit;
70 maintainers = with lib.maintainers; [ bcdarwin ];
71 broken = true; # Nixpkgs missing `albumentations` and `prodict`
72 };
73}