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