python3Packages.orbax-checkpoint: 0.11.26 -> 0.11.27 (#458865)

buildPythonPackage rec {

  pname = "orbax-checkpoint";

  version = "0.11.26";

  version = "0.11.27";

  pyproject = true;

  src = fetchFromGitHub {

    owner = "google";

    repo = "orbax";

    tag = "v${version}";

    hash = "sha256-CY5Bs/o8fU57QJETYnyJVkP7Y+cahpqZftyIJNU+GvU=";

    hash = "sha256-KXEwUHJAFxeJTs3qKUXMCYO3t5hoPTjGay9GIOQIuCY=";

  };

  sourceRoot = "${src.name}/checkpoint";

    # AssertionError: 2 not greater than 2.0046136379241943

    "test_async_mkdir_parallel"

    "test_async_mkdir_sequential"

    # AssertionError:

    # "Handler type string "(?:__main__|orbax.checkpoint._src.handlers.handler_type_registry_test)\.TestHandler" not found in the registry."

    # does not match

    # "'Handler type string "handler_type_registry_test.TestHandler" not found in the registry.'"

    "test_get_handler_type_not_found"

    "test_no_typestr"

    "test_register_duplicate_handler_type"

    # AssertionError: False is not true

    "test_register_and_get"

    "test_register_different_modules"

  ]

  ++ lib.optionals stdenv.hostPlatform.isDarwin [

    # Probably failing because of a filesystem impurity

    # First from multiprocess_test, Second from orbax.checkpoint._src.testing.multiprocess_test.

    # Description from first occurrence: Number of processes to use.

    # https://github.com/google/orbax/issues/1580

    "orbax/checkpoint/_src/testing/multiprocess_test.py"

    "orbax/checkpoint/experimental/emergency/"

    # ValueError: Distributed system is not available; please initialize it via `jax.distributed.initialize()` at the start of your program.

    "orbax/checkpoint/_src/handlers/array_checkpoint_handler_test.py"

    # import file mismatch:

    # imported module 'registry_test' has this __file__ attribute:

    #   /build/source/checkpoint/orbax/checkpoint/experimental/v1/_src/layout/registry_test.py

    # which is not the same as the test file we want to collect:

    #   /build/source/checkpoint/orbax/checkpoint/experimental/v1/_src/serialization/registry_test.py

    # HINT: remove __pycache__ / .pyc files and/or use a unique basename for your test file module

    "orbax/checkpoint/experimental/v1/_src/serialization/registry_test.py"

    # E   FileNotFoundError: [Errno 2] No such file or directory:

    # '/build/absl_testing/DefaultSnapshotTest/runTest/root/path/to/source/data.txt'

    "orbax/checkpoint/_src/path/snapshot/snapshot_test.py"

    "orbax/checkpoint/_src/tree/parts_of_test.py"

    "orbax/checkpoint/_src/tree/structure_utils_test.py"

    "orbax/checkpoint/_src/tree/utils_test.py"

    "orbax/checkpoint/checkpoint_manager_test.py"

    "orbax/checkpoint/single_host_test.py"

    "orbax/checkpoint/transform_utils_test.py"

  ];