code: def maybe_get_ax(*args, **kwargs):
      
  
      if 'ax' in kwargs:
          ax = kwargs.pop('ax')
      elif len(args) == 0:
          fig = plt.gcf()
          ax = plt.gca()
      elif isinstance(args[0], mpl.axes.Axes):
          ax = args[0]
          args = args[1:]
      else:
          ax = plt.gca()
      return ax, args, dict(kwargs)
docs: It used to be that the first argument of prettyplotlib had to be the 'ax'
      object, but that's not the case anymore.
  
      @param args:
      @type args:
      @param kwargs:
      @type kwargs:
      @return:
      @rtype: