diff options
author | Samuel Ainsworth <skainsworth@gmail.com> | 2022-02-21 15:33:36 -0800 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-02-21 15:33:36 -0800 |
commit | cf989b2680920d62c2bf89900a6080dbd828e5c4 (patch) | |
tree | 686d47892c68f183342273815a175954f7289a4a /pkgs | |
parent | d9a26dd8772fd3b056da3e4ea1f759fe22d46270 (diff) | |
parent | 33b6c73fc3319f1a32941cb653f856c163bdb4ca (diff) |
Merge pull request #160946 from samuela/samuela/augmax
python3Packages.augmax: init at unstable-2022-02-19
Diffstat (limited to 'pkgs')
-rw-r--r-- | pkgs/development/python-modules/augmax/default.nix | 36 | ||||
-rw-r--r-- | pkgs/top-level/python-packages.nix | 2 |
2 files changed, 38 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/augmax/default.nix b/pkgs/development/python-modules/augmax/default.nix new file mode 100644 index 0000000000000..cda3ff5fbcd23 --- /dev/null +++ b/pkgs/development/python-modules/augmax/default.nix @@ -0,0 +1,36 @@ +{ buildPythonPackage +, einops +, fetchFromGitHub +, jax +, jaxlib +, lib +}: + +buildPythonPackage rec { + pname = "augmax"; + version = "unstable-2022-02-19"; + format = "setuptools"; + + src = fetchFromGitHub { + owner = "khdlr"; + repo = pname; + # augmax does not have releases tagged. See https://github.com/khdlr/augmax/issues/5. + rev = "3e5d85d6921a1e519987d33f226bc13f61e04d04"; + sha256 = "046n43v7161w7najzlbi0443q60436xv24nh1mv23yw6psqqhx5i"; + }; + + propagatedBuildInputs = [ einops jax ]; + + # augmax does not have any tests at the time of writing (2022-02-19), but + # jaxlib is necessary for the pythonImportsCheckPhase. + checkInputs = [ jaxlib ]; + + pythonImportsCheck = [ "augmax" ]; + + meta = with lib; { + description = "Efficiently Composable Data Augmentation on the GPU with Jax"; + homepage = "https://github.com/khdlr/augmax"; + license = licenses.asl20; + maintainers = with maintainers; [ samuela ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index b44d123bed28b..664c196a659bf 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -760,6 +760,8 @@ in { inherit (pkgs) augeas; }; + augmax = callPackage ../development/python-modules/augmax { }; + auroranoaa = callPackage ../development/python-modules/auroranoaa { }; aurorapy = callPackage ../development/python-modules/aurorapy { }; |