human
Can you provide a the Python code you requested given the following:
Complete the below
def keras_test(func):

    ''''Function wrapper to clean up after TensorFlow tests.
    # Arguments
    func: test function to clean up after.
    # Returns
    A function wrapping the input function.'
    '''
assistant
#Here is the Python code you requested.
    @six.wraps(func)
    def wrapper(*args, **kwargs):
        output = func(*args, **kwargs)
        if (K.backend() == 'tensorflow'):
            K.clear_session()
        return output
    return wrapper