Gaetan Lepage 2025-06-10 21:49:04 +02:00
parent f824d91fdf
commit 988601eb6f

View File

@ -12,7 +12,6 @@
jax, jax,
jaxlib, jaxlib,
numpy, numpy,
etils,
# tests # tests
callPackage, callPackage,
@ -20,14 +19,14 @@
buildPythonPackage rec { buildPythonPackage rec {
pname = "optax"; pname = "optax";
version = "0.2.4"; version = "0.2.5";
pyproject = true; pyproject = true;
src = fetchFromGitHub { src = fetchFromGitHub {
owner = "deepmind"; owner = "deepmind";
repo = "optax"; repo = "optax";
tag = "v${version}"; tag = "v${version}";
hash = "sha256-7UPWeo/Q9/tjewaM7HN8/e7U1U1QzAliuk95+9GOi0E="; hash = "sha256-EGQeRYSxHdENqB3QPZFsjqwh4LYT5CF8E1K3fKFedPg=";
}; };
outputs = [ outputs = [
@ -40,11 +39,10 @@ buildPythonPackage rec {
dependencies = [ dependencies = [
absl-py absl-py
chex chex
etils
jax jax
jaxlib jaxlib
numpy numpy
] ++ etils.optional-dependencies.epy; ];
postInstall = '' postInstall = ''
mkdir $testsout mkdir $testsout