Use `ocp.future.CommitFutureAwaitingContractedSignals` in custom async checkpoint handlers and type handlers to support async directory creation.#1737

Open

copybara-service[bot] wants to merge 1 commit intomainfrom test_740001910

Commits on Mar 25, 2025

Use ocp.future.CommitFutureAwaitingContractedSignals in custom async checkpoint handlers and type handlers to support async directory creation.

mridul-sahu
authored and
Orbax Authors
committed