Add strict mode support in chex.assert_trees_all_close. This aligns the behaviour with chex.assert_trees_all_equal.#436

copybara-service[bot] merged 1 commit intomaingoogle-deepmind/chex:mainfrom

test_862181409google-deepmind/chex:test_862181409Copy head branch name to clipboard