Fix race condition in chexify#271

copybara-service[bot] wants to merge 1 commit intomaingoogle-deepmind/chex:mainfrom

test_534012863google-deepmind/chex:test_534012863Copy head branch name to clipboard