diff --git a/pkgs/development/python-modules/batchgenerators/default.nix b/pkgs/development/python-modules/batchgenerators/default.nix new file mode 100644 index 00000000000..001c3cc19a5 --- /dev/null +++ b/pkgs/development/python-modules/batchgenerators/default.nix @@ -0,0 +1,38 @@ +{ lib +, buildPythonPackage +, isPy27 +, fetchPypi +, pytest +, unittest2 +, future +, numpy +, scipy +, scikitlearn +, scikitimage +, threadpoolctl +}: + + +buildPythonPackage rec { + pname = "batchgenerators"; + version = "0.19.7"; + + disabled = isPy27; + + src = fetchPypi { + inherit pname version; + sha256 = "0qqzwqf5r0q6jh8avz4f9kf8x96crvdnkznhf24pbm0faf8yk67q"; + }; + + propagatedBuildInputs = [ future numpy scipy scikitlearn scikitimage threadpoolctl ]; + checkInputs = [ pytest unittest2 ]; + + checkPhase = "pytest tests"; + + meta = { + description = "2D and 3D image data augmentation for deep learning"; + homepage = "https://github.com/MIC-DKFZ/batchgenerators"; + license = lib.licenses.asl20; + maintainers = with lib.maintainers; [ bcdarwin ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index e1ea2c9d1e0..7c51a3feedb 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -1803,6 +1803,8 @@ in { base58 = callPackage ../development/python-modules/base58 {}; + batchgenerators = callPackage ../development/python-modules/batchgenerators { }; + batinfo = callPackage ../development/python-modules/batinfo {}; bcdoc = callPackage ../development/python-modules/bcdoc {};