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