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

python3Packages.orbax-checkpoint: 0.11.28 -> 0.11.30 (#465625)

Browse Source
This commit is contained in:
kirillrdy
2025-11-27 20:02:34 +00:00
committed by GitHub
parent ffe72c37cd 035ae46b0f
commit 4339f29144
2 changed files with 39 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkgs/development/python-modules/orbax-checkpoint/default.nix
+24 -6
View File
@@ -33,28 +33,25 @@
pytest-xdist,
pytestCheckHook,
safetensors,
torch,
}:
buildPythonPackage rec {
pname = "orbax-checkpoint";
version = "0.11.28";
version = "0.11.30";
pyproject = true;
src = fetchFromGitHub {
owner = "google";
repo = "orbax";
tag = "v${version}";
hash = "sha256-a7E60fZRmEXTA220mwr7EDMUc+zYbW7wG40vY7NeAOM=";
hash = "sha256-y8l0AVGt2t5zLX+x+yuWHsEDy68agpXIkrew+zfYGXU=";
};
sourceRoot = "${src.name}/checkpoint";
build-system = [ flit-core ];
pythonRelaxDeps = [
"jax"
];
dependencies = [
absl-py
aiofiles
@@ -82,6 +79,7 @@ buildPythonPackage rec {
pytest-xdist
pytestCheckHook
safetensors
torch
];
pythonImportsCheck = [
@@ -115,6 +113,26 @@ buildPythonPackage rec {
];
disabledTestPaths = [
# import file mismatch:
# imported module 'sharding_test' has this __file__ attribute:
# /build/source/checkpoint/orbax/checkpoint/_src/arrays/sharding_test.py
# which is not the same as the test file we want to collect:
# /build/source/checkpoint/orbax/checkpoint/_src/metadata/sharding_test.py
"orbax/checkpoint/_src/metadata/sharding_test.py"
# Circular dependency with clu
"orbax/checkpoint/_src/testing/benchmarks/array_handler_benchmark_test.py"
"orbax/checkpoint/_src/testing/benchmarks/checkpoint_manager_benchmark_test.py"
"orbax/checkpoint/_src/testing/benchmarks/checkpoint_manager_perf_benchmark_test.py"
"orbax/checkpoint/_src/testing/benchmarks/checkpoint_policy_benchmark_test.py"
"orbax/checkpoint/_src/testing/benchmarks/core/config_parsing_test.py"
"orbax/checkpoint/_src/testing/benchmarks/core/core_test.py"
"orbax/checkpoint/_src/testing/benchmarks/core/metric_test.py"
"orbax/checkpoint/_src/testing/benchmarks/emergency_checkpoint_manager_benchmark_test.py"
"orbax/checkpoint/_src/testing/benchmarks/multihost_dispatchers_benchmark_test.py"
"orbax/checkpoint/_src/testing/benchmarks/pytree_checkpoint_benchmark_test.py"
"orbax/checkpoint/_src/testing/benchmarks/single_replica_benchmark_test.py"
# E absl.flags._exceptions.DuplicateFlagError: The flag 'num_processes' is defined twice.
# First from multiprocess_test, Second from orbax.checkpoint._src.testing.multiprocess_test.
# Description from first occurrence: Number of processes to use.
pkgs/development/python-modules/tensorstore/default.nix
+15 -15
View File
@@ -12,28 +12,28 @@
let
pythonVersionNoDot = builtins.replaceStrings [ "." ] [ "" ] python.pythonVersion;
systemToPlatform = {
"aarch64-linux" = "manylinux_2_17_aarch64.manylinux2014_aarch64";
"x86_64-linux" = "manylinux_2_17_x86_64.manylinux2014_x86_64";
"aarch64-linux" = "manylinux_2_27_aarch64.manylinux_2_28_aarch64";
"x86_64-linux" = "manylinux_2_27_x86_64.manylinux_2_28_x86_64";
"aarch64-darwin" = "macosx_11_0_arm64";
};
hashes = {
"310-x86_64-linux" = "sha256-GmzcxS5LhB0j5Qovoo4Bbm2fYdbqkYjUVV6hibBAoPY=";
"311-x86_64-linux" = "sha256-NzVYuAPYwsV/xhOxEAeuWBOfGaPN3UQ6DeXXtTIeWWE=";
"312-x86_64-linux" = "sha256-ztVDC836f8s6a9xEczF2FYy4d7Nb3SM8rILiW0zJTpI=";
"313-x86_64-linux" = "sha256-UrVG8Hayw78hfGDwXeQSTMEZfOkvjoJufsc64yQHSlo=";
"310-aarch64-linux" = "sha256-lQQbVaLshtH2aQUS0Yg1gbGPL09Gw9l4lK6wrC22r38=";
"311-aarch64-linux" = "sha256-ZcOhoqNaG1N0A/NkA9JYyqtHflZLwPZBCblBzHe08gM=";
"312-aarch64-linux" = "sha256-h6l6NLBHXdx9KvxA5d1/jRJSKqge37zMs5Yoz1kUVNU=";
"313-aarch64-linux" = "sha256-YnbiebRetdm5XE3z55ViVfQU/UsSjS3hbYrs3obDY1c";
"310-aarch64-darwin" = "sha256-uWG7u3ocakjkwUBqmMrr60AEYeLnWgi23wwBMpQDehU=";
"311-aarch64-darwin" = "sha256-9A5zvNwzPfs/f+D88CO8vsQVM8mFZldxj/duzhoZAuA=";
"312-aarch64-darwin" = "sha256-06JP62GV8cIiFillwBB8n/VtMizKI+GfDmZjb264DxQ=";
"313-aarch64-darwin" = "sha256-3ohD+zRiiZ3nvN7qzLkjA6nWEAa8NjZN60qI30YyC6Q=";
"311-x86_64-linux" = "sha256-gwcu4OVR1tylguFUtkyLgGbSduwHWXhOMUnCghKmHxg=";
"312-x86_64-linux" = "sha256-YI9xeOxuTko8JlRbCkT0S/g0ONBL8tlgzQ52meqpnvY=";
"313-x86_64-linux" = "sha256-yfLcM0LkaGr5j24lncn7N38b9le2ScJHv2ZHu+T5gJA=";
"314-x86_64-linux" = "sha256-evlCImnCv83s+d1VMJBgZlq5wtf2yJI3ftMsAyQA/uo=";
"311-aarch64-linux" = "sha256-xCMLj9KXleiORB90nYgZc+yo2t8zxSYrNng5+4iR95s=";
"312-aarch64-linux" = "sha256-3r1DUELAC+aLofs89ZMlp7q7P0o89HRMh93jRoAsu7Q=";
"313-aarch64-linux" = "sha256-lNj8nfFyGwKHBGrKcgn9UECInK1CAue3Oh/bd82bccY=";
"314-aarch64-linux" = "sha256-hHmCZSJz+3staUt4kgV0eq8+UK5kc4xct7XrA9hqmUc=";
"311-aarch64-darwin" = "sha256-XhUtM0vzT7q9/o5bw1uH0fmUcGWST/g8KeZZMIs26Ug=";
"312-aarch64-darwin" = "sha256-EIwOhnqiyH1JgsxjJaLeDE9b1jwr6hitsZOjcMQFlM4=";
"313-aarch64-darwin" = "sha256-Kc9DNhU68TasisUo4u1G3xk2ftrn4U43vKGot8SEjvI=";
"314-aarch64-darwin" = "sha256-l3VtLLo8XOIeFWAsKvWgJSHMDs2n+fttGNovO9UYJ/Q=";
};
in
buildPythonPackage rec {
pname = "tensorstore";
version = "0.1.71";
version = "0.1.79";
format = "wheel";
# The source build involves some wonky Bazel stuff.