def time_me(fn):
    """
    Wrapper for testing the function time
    Args:
        fn: function

    Returns: None

    """
    def _wrapper(*args, **kwargs):
        start = time.time()
        ret = fn(*args, **kwargs)
        print("function: %s cost %s second" % (fn.__name__, time.time() - start))
        return ret

    return _wrapper

    Wrapper for testing the function time
    Args:
        fn: function

    Returns: None

    
metadrive/utils/utils.py