···11{ buildPythonPackage
22, einops
33-, fetchPypi
33+, fetchFromGitHub
44, jax
55, jaxlib
66, lib
···15151616 disbaled = pythonOlder "3.6";
17171818- # Using fetchPypi because the latest version was not tagged on GitHub.
1919- # Switch back to fetchFromGitHub when a tag will be available
2020- # https://github.com/khdlr/augmax/issues/8
2121- src = fetchPypi {
2222- inherit pname version;
2323- hash = "sha256-pf1DTaHA7D+s2rqwwGYlJrJOI7fok+WOvOCtZhOOGHo=";
1818+ src = fetchFromGitHub {
1919+ owner = "khdlr";
2020+ repo = "augmax";
2121+ rev = "refs/tags/v${version}";
2222+ hash = "sha256-xz6yJiVZUkRcRa2rKZdytfpP+XCk/QI4xtKlNaS9FYo=";
2423 };
25242626- nativeBuildInputs = [
2525+ build-system = [
2726 setuptools
2827 ];
29283030- propagatedBuildInputs = [ einops jax ];
2929+ dependencies = [
3030+ einops
3131+ jax
3232+ ];
31333234 # augmax does not have any tests at the time of writing (2022-02-19), but
3335 # jaxlib is necessary for the pythonImportsCheckPhase.