      def lines_sort(self, keys):
          """ Sort the data keys into consistent order
              and set line color map and line width """
          logger.trace("Sorting lines")
          raw_lines = list()
          sorted_lines = list()
          for key in sorted(keys):
              title = key.replace("_", " ").title()
              if key.startswith("raw"):
                  raw_lines.append([key, title])
              else:
 
</s> remove     def lines_sort(self, keys):
        """ Sort the data keys into consistent order
            and set line color map and line width """
 </s> add         Parameters
        ----------
        keys: list
            The list of data point keys

        Returns
        -------
        list
            A list of loss keys with their corresponding line formatting and color information
        """ </s> remove         self.ax1.set_yscale(scale)
 </s> add         self._ax1.set_yscale(scale)

    def _lines_sort(self, keys: List[str]) -> List[List[Union[str, int, Tuple[float]]]]:
        """ Sort the data keys into consistent order and set line color map and line width. </s> remove     def lines_style(self, lines, groupsize):
        """ Set the color map and line width for each group """
 </s> add     def _lines_style(self,
                     lines: List[str],
                     groupsize: int) -> List[List[Union[str, int, Tuple[float]]]]:
        """ Obtain the color map and line width for each group.

        Parameters
        ----------
        lines: list
            The list of sorted line keys to display on the graph
        groupsize: int
            The size of each group to display in the graph

        Returns
        -------
        list
            A list of loss keys with their corresponding line formatting and color information
        """ </s> remove         groupsize = self.lines_groupsize(raw_lines, sorted_lines)
 </s> add         groupsize = self._lines_groupsize(raw_lines, sorted_lines) </s> remove     def axes_set_yscale(self, scale):
        """ Set the Y-Scale to log or linear """
 </s> add     def _axes_set_yscale(self, scale: str) -> None:
        """ Set the Y-Scale to log or linear

        Parameters
        ----------
        scale: str
            Should be one of ``"log"`` or ``"linear"``
        """
        raw_lines = []
        sorted_lines = []
      def lines_sort(self, keys):
          """ Sort the data keys into consistent order
              and set line color map and line width """
          logger.trace("Sorting lines")
        raw_lines = []
        sorted_lines = []
        raw_lines = []
        sorted_lines = []
          for key in sorted(keys):
              title = key.replace("_", " ").title()
              if key.startswith("raw"):
                  raw_lines.append([key, title])
              else:
keep keep keep keep replace replace keep keep keep keep keep
 Update matplotlib requirement
  - bump min requirement to 3.5.1
  - Fix custom toolbar for v3.5.1
  - Documentation + linting 
lib/gui/display_graph.py