Revert to apple's patch for testing.#2401

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

test_808647442google/orbax:test_808647442Copy head branch name to clipboard