···1{ buildPythonPackage
2, einops
3-, fetchPypi
4, jax
5, jaxlib
6, lib
···1516 disbaled = pythonOlder "3.6";
1718- # Using fetchPypi because the latest version was not tagged on GitHub.
19- # Switch back to fetchFromGitHub when a tag will be available
20- # https://github.com/khdlr/augmax/issues/8
21- src = fetchPypi {
22- inherit pname version;
23- hash = "sha256-pf1DTaHA7D+s2rqwwGYlJrJOI7fok+WOvOCtZhOOGHo=";
24 };
2526- nativeBuildInputs = [
27 setuptools
28 ];
2930- propagatedBuildInputs = [ einops jax ];
0003132 # augmax does not have any tests at the time of writing (2022-02-19), but
33 # jaxlib is necessary for the pythonImportsCheckPhase.
···1{ buildPythonPackage
2, einops
3+, fetchFromGitHub
4, jax
5, jaxlib
6, lib
···1516 disbaled = pythonOlder "3.6";
1718+ src = fetchFromGitHub {
19+ owner = "khdlr";
20+ repo = "augmax";
21+ rev = "refs/tags/v${version}";
22+ hash = "sha256-xz6yJiVZUkRcRa2rKZdytfpP+XCk/QI4xtKlNaS9FYo=";
023 };
2425+ build-system = [
26 setuptools
27 ];
2829+ dependencies = [
30+ einops
31+ jax
32+ ];
3334 # augmax does not have any tests at the time of writing (2022-02-19), but
35 # jaxlib is necessary for the pythonImportsCheckPhase.