def across_data_formats(func):
@six.wraps(func)<EOL>def wrapper(*args, **kwargs):<EOL><INDENT>for data_format in {'<STR_LIT>', '<STR_LIT>'}:<EOL><INDENT>K.set_image_data_format(data_format)<EOL>func(*args, **kwargs)<EOL>if K.backend() == '<STR_LIT>':<EOL><INDENT>K.clear_session()<EOL>tf.reset_default_graph()<EOL><DEDENT><DEDENT><DEDENT>return wrapper<EOL>
Function wrapper to run tests on multiple keras data_format and clean up after TensorFlow tests.

    Args:
        func: test function to clean up after.

    Returns:
        A function wrapping the input function.