nixpkgs/pkgs/development/python-modules/rising/default.nix
2022-09-04 14:13:52 +02:00

38 lines
783 B
Nix

{ lib
, buildPythonPackage
, isPy27
, fetchFromGitHub
, pytestCheckHook
, dill
, numpy
, torch
, threadpoolctl
, tqdm
}:
buildPythonPackage rec {
pname = "rising";
version = "0.2.1";
disabled = isPy27;
src = fetchFromGitHub {
owner = "PhoenixDL";
repo = pname;
rev = "v${version}";
sha256 = "15wYWToXRae1cMpHWbJwzAp0THx6ED9ixQgL+n1v9PI=";
};
propagatedBuildInputs = [ numpy torch threadpoolctl tqdm ];
checkInputs = [ dill 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 ];
};
}