colin/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #311296 from GaetanLepage/augmax

Browse Source 
python311Packages.augmax: switch to fetchFromGitHub
This commit is contained in:
Pol Dellaiera 2024-05-13 11:57:43 +02:00 committed by GitHub
commit 055b329680
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
pkgs/development/python-modules/augmax/default.nix
View File

@ -1,6 +1,6 @@
{ buildPythonPackage
, einops
, fetchPypi
, fetchFromGitHub
, jax
, jaxlib
, lib
@ -15,19 +15,21 @@ buildPythonPackage rec {
disbaled = pythonOlder "3.6";
# Using fetchPypi because the latest version was not tagged on GitHub.
# Switch back to fetchFromGitHub when a tag will be available
# https://github.com/khdlr/augmax/issues/8
src = fetchPypi {
inherit pname version;
hash = "sha256-pf1DTaHA7D+s2rqwwGYlJrJOI7fok+WOvOCtZhOOGHo=";
src = fetchFromGitHub {
owner = "khdlr";
repo = "augmax";
rev = "refs/tags/v${version}";
hash = "sha256-xz6yJiVZUkRcRa2rKZdytfpP+XCk/QI4xtKlNaS9FYo=";
};
nativeBuildInputs = [
build-system = [
setuptools
];
propagatedBuildInputs = [ einops jax ];
dependencies = [
einops
jax
];
# augmax does not have any tests at the time of writing (2022-02-19), but
# jaxlib is necessary for the pythonImportsCheckPhase.