pyrox.dev
1{ lib
2, buildPythonPackage
3, isPy27
4, fetchFromGitHub
5, pytestCheckHook
6, pytest-cov
7, dill
8, numpy
9, pytorch
10, threadpoolctl
11, tqdm
12}:
13
14buildPythonPackage rec {
15 pname = "rising";
16 version = "0.2.1";
17
18 disabled = isPy27;
19
20 src = fetchFromGitHub {
21 owner = "PhoenixDL";
22 repo = pname;
23 rev = "v${version}";
24 sha256 = "15wYWToXRae1cMpHWbJwzAp0THx6ED9ixQgL+n1v9PI=";
25 };
26
27 propagatedBuildInputs = [ numpy pytorch threadpoolctl tqdm ];
28 checkInputs = [ dill pytest-cov pytestCheckHook ];
29
30 disabledTests = [ "test_affine" ]; # deprecated division operator '/'
31
32 meta = {
33 description = "High-performance data loading and augmentation library in PyTorch";
34 homepage = "https://rising.rtfd.io";
35 license = lib.licenses.mit;
36 maintainers = with lib.maintainers; [ bcdarwin ];
37 };
38}