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

Merge pull request #280141 from GaetanLepage/equinox

Browse Source 
python311Packages.equinox: 0.11.2 -> 0.11.3
This commit is contained in:
OTABI Tomoya 2024-01-11 21:14:22 +09:00 committed by GitHub
commit 20451a09ca
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
pkgs/development/python-modules/equinox/default.nix
View File

@ -1,7 +1,7 @@
{ lib
, buildPythonPackage
, pythonOlder
, fetchFromGitHub
, fetchpatch
, hatchling
, jax
, jaxlib
@ -9,29 +9,24 @@
, typing-extensions
, beartype
, optax
, pytest-xdist
, pytestCheckHook
}:
buildPythonPackage rec {
pname = "equinox";
version = "0.11.2";
version = "0.11.3";
pyproject = true;
disabled = pythonOlder "3.9";
src = fetchFromGitHub {
owner = "patrick-kidger";
repo = "equinox";
rev = "refs/tags/v${version}";
hash = "sha256-qFTKiY/t2LCCWJBOSfaX0hYQInrpXgfhTc+J4iuyVbM=";
hash = "sha256-la3gPfwQ2pxfZoEikn9uG+Pc3PKafgEgxZ8oVQEm9YM=";
};
patches = [
(fetchpatch { # https://github.com/patrick-kidger/equinox/pull/601
name = "fix-wrong-PRNGKey-annotation";
url = "https://github.com/patrick-kidger/equinox/pull/601/commits/dce2fa1b7dcfd25d9573ce3186c5f6e8f79392bb.patch";
hash = "sha256-tlGV5xuNGLZTd1GlPwllybPz8tWHGHaCBdlsEuISm/0=";
})
];
nativeBuildInputs = [
hatchling
];
@ -46,6 +41,7 @@ buildPythonPackage rec {
nativeCheckInputs = [
beartype
optax
pytest-xdist
pytestCheckHook
];
@ -53,6 +49,7 @@ buildPythonPackage rec {
meta = with lib; {
description = "A JAX library based around a simple idea: represent parameterised functions (such as neural networks) as PyTrees";
changelog = "https://github.com/patrick-kidger/equinox/releases/tag/v${version}";
homepage = "https://github.com/patrick-kidger/equinox";
license = licenses.asl20;
maintainers = with maintainers; [ GaetanLepage ];