pyrox.dev
1{ buildPythonPackage
2, fetchFromGitHub
3, pythonOlder
4, pytest
5, h5py
6, hickle
7, numpy
8, pandas
9, pillow
10, six
11, pytorch
12, torchvision
13, tqdm
14, lib
15}:
16
17buildPythonPackage rec {
18 pname = "pywick";
19 version = "0.5.6";
20
21 disabled = pythonOlder "3.6";
22
23 src = fetchFromGitHub {
24 owner = "achaiah";
25 repo = pname;
26 rev = "v${version}";
27 sha256 = "1gmlifnv9kji0d1jwg1pa8d96zg48w17qg0sgxwy1y1jf3hn37bm";
28 };
29
30 propagatedBuildInputs = [
31 h5py hickle numpy pandas pillow six pytorch torchvision tqdm
32 ];
33
34 checkInputs = [ pytest ];
35
36 checkPhase = ''
37 runHook preCheck
38 pytest tests/
39 runHook postCheck
40 '';
41
42 meta = {
43 description = "High-level training framework for Pytorch";
44 homepage = "https://github.com/achaiah/pywick";
45 license = lib.licenses.mit;
46 maintainers = with lib.maintainers; [ bcdarwin ];
47 };
48}