pyrox.dev
1{
2 stdenv,
3 lib,
4 buildPythonPackage,
5 pythonOlder,
6 fetchFromGitHub,
7 pytestCheckHook,
8 pythonRelaxDepsHook,
9 dill,
10 lightning-utilities,
11 numpy,
12 torch,
13 threadpoolctl,
14 tqdm,
15}:
16
17buildPythonPackage rec {
18 pname = "rising";
19 version = "0.3.0";
20 format = "setuptools";
21 disabled = pythonOlder "3.8";
22
23 src = fetchFromGitHub {
24 owner = "PhoenixDL";
25 repo = pname;
26 rev = "refs/tags/v${version}";
27 hash = "sha256-sBzVTst5Tp2oZZ+Xsg3M7uAMbucL6idlpYwHvib3EaY=";
28 };
29
30 nativeBuildInputs = [ pythonRelaxDepsHook ];
31
32 pythonRelaxDeps = [ "lightning-utilities" ];
33
34 propagatedBuildInputs = [
35 lightning-utilities
36 numpy
37 torch
38 threadpoolctl
39 tqdm
40 ];
41 nativeCheckInputs = [
42 dill
43 pytestCheckHook
44 ];
45 disabledTests = lib.optionals (stdenv.isLinux && stdenv.isAarch64) [
46 # RuntimeError: DataLoader worker (pid(s) <...>) exited unexpectedly:
47 "test_progressive_resize_integration"
48 ];
49
50 pythonImportsCheck = [
51 "rising"
52 "rising.loading"
53 "rising.ops"
54 "rising.random"
55 "rising.transforms"
56 "rising.transforms.functional"
57 "rising.utils"
58 ];
59
60 meta = {
61 description = "High-performance data loading and augmentation library in PyTorch";
62 homepage = "https://rising.rtfd.io";
63 license = lib.licenses.mit;
64 maintainers = with lib.maintainers; [ bcdarwin ];
65 };
66}