Skip to content

Commit

Permalink
python312Packages.orbax-checkpoint: 0.6.4 -> 0.10.2
Browse files Browse the repository at this point in the history
  • Loading branch information
GaetanLepage committed Dec 4, 2024
1 parent 2f9d395 commit 0f076c0
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions pkgs/development/python-modules/orbax-checkpoint/default.nix
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@
numpy,
protobuf,
pyyaml,
simplejson,
tensorstore,
typing-extensions,

Expand All @@ -32,14 +33,14 @@

buildPythonPackage rec {
pname = "orbax-checkpoint";
version = "0.6.4";
version = "0.10.2";
pyproject = true;

src = fetchFromGitHub {
owner = "google";
repo = "orbax";
rev = "refs/tags/v${version}";
hash = "sha256-xd75/AKBFUdA6a8sQnCB2rVbHl/Foy4LTb07jnwrTjA=";
hash = "sha256-jGo0V2M1xS0a1GMFHLxxilNvlgr12RWodvFRJwcMsm8=";
};

sourceRoot = "${src.name}/checkpoint";
Expand All @@ -58,6 +59,7 @@ buildPythonPackage rec {
numpy
protobuf
pyyaml
simplejson
tensorstore
typing-extensions
];
Expand Down Expand Up @@ -91,7 +93,7 @@ buildPythonPackage rec {
meta = {
description = "Orbax provides common utility libraries for JAX users";
homepage = "https://github.com/google/orbax/tree/main/checkpoint";
changelog = "https://github.com/google/orbax/releases/tag/v${version}";
changelog = "https://github.com/google/orbax/blob/${lib.removePrefix "refs/tags/" src.rev}/checkpoint/CHANGELOG.md";
license = lib.licenses.asl20;
maintainers = with lib.maintainers; [ fab ];
};
Expand Down

0 comments on commit 0f076c0

Please sign in to comment.