def retry_if_exception_type()

in python/pipelines/components/bigquery/component.py [0:0]

• 9 lines of code
• 2 McCabe index (conditional complexity)

    def retry_if_exception_type(exception_types):
        def decorator(func):
            def new_func(*args, **kwargs):
                try:
                    return func(*args, **kwargs)
                except exception_types as error:
                    raise exceptions.RetryError(error, cause=error)
            return new_func
        return decorator