From fbe31e0739a9ffdff0c493bc556d588b3310a565 Mon Sep 17 00:00:00 2001 From: Gaetan Lepage Date: Thu, 27 Nov 2025 20:22:16 +0000 Subject: [PATCH] python3Packages.orbax-checkpoint: skip all benchmark tests --- .../python-modules/orbax-checkpoint/default.nix | 14 ++------------ 1 file changed, 2 insertions(+), 12 deletions(-) diff --git a/pkgs/development/python-modules/orbax-checkpoint/default.nix b/pkgs/development/python-modules/orbax-checkpoint/default.nix index e39356a36402..2a367a350d93 100644 --- a/pkgs/development/python-modules/orbax-checkpoint/default.nix +++ b/pkgs/development/python-modules/orbax-checkpoint/default.nix @@ -120,18 +120,8 @@ buildPythonPackage rec { # /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" + # Circular dependency with clu (and we should not run benchmarks anyway) + "orbax/checkpoint/_src/testing/benchmarks/" # E absl.flags._exceptions.DuplicateFlagError: The flag 'num_processes' is defined twice. # First from multiprocess_test, Second from orbax.checkpoint._src.testing.multiprocess_test.