pkgs/development/python-modules/rising/default.nix

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