code: def customize(func):
      
      @wraps(func)
      def call_w_context(*args, **kwargs):
          set_context = kwargs.pop('set_context', True)
          if set_context:
              with plotting_context(), axes_style():
                  return func(*args, **kwargs)
          else:
              return func(*args, **kwargs)
      return call_w_context
docs: Decorator to set plotting context and axes style during function call.