Add tree-mode arg error handling#5251

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

test_869190035google/flax:test_869190035Copy head branch name to clipboard