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

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