Display test execution mode in stack trace.
Previously when using `run_all_keras_modes` in tests, the execution mode was not included in the stack trace, making it difficult to debug issues. Now the execution mode is clearly printed out in the stack trace (it's the name of the last function call before the actual test execution). PiperOrigin-RevId: 226079310
Loading
Please sign in to comment