docs: def plot(self, ax, scatter=True, grid=True, progress_colors=True, progress_max=1., depth=10, plot_dims=[0,1]):
          """
          Plot a projection on 2D of the Tree.
          
          Parameters
          ----------
          ax : plt axis
          scatter : bool
              If the points are ploted
          grid : bool
              If the leaves' bounds are ploted
          progress_colors : bool
              If rectangles are filled with colors based on progress 
          progress_max : float
              Max progress on color scale (will be ploted as 1.)
          depth : int
              Max depth of the ploted nodes
          plot_dims : list
              List of the 2 dimensions to project tree on
          
          """
          
          if scatter and self.get_data_x() is not None:
              self.plot_scatter(ax, plot_dims)
              
          if grid:
              self.plot_grid(ax, progress_colors, progress_max, depth, plot_dims)