Added test to check for Variable warning.#2610

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

chiamp:test_returned_variable_warningchiamp/flax:test_returned_variable_warningCopy head branch name to clipboard