Use a separate _is_saving_in_progress bool to track whether a save is ongoing, as this prevents conflicts when concurrent wait_until_finished calls cause the finalize thread lock to block.
#16077
copybara-service[bot]build.yml
on: pull_request
Matrix: build-checkpoint
Matrix: build-export
Matrix: build-orbax-model
Annotations
9 errors and 11 warnings