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