  
      @staticmethod
      def lines_groupsize(raw_lines, sorted_lines):
          """ Get the number of items in each group.
              If raw data isn't selected, then check the length of
              remaining groups until something is found """
          groupsize = 1
          if raw_lines:
              groupsize = len(raw_lines)
          elif sorted_lines:
              keys = [key[0][:key[0].find("_")] for key in sorted_lines]
 
</s> remove     def lines_groupsize(raw_lines, sorted_lines):
 </s> add     def _lines_groupsize(raw_lines: List[str], sorted_lines: List[str]) -> int: </s> remove         lines = self.lines_style(sorted_lines, groupsize)
 </s> add         lines = self._lines_style(sorted_lines, groupsize) </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     def lines_create_colors(self, groupsize, groups):
        """ Create the colors """
        colours = list()
 </s> add     def _lines_create_colors(self, groupsize: int, groups: int) -> List[Tuple[float]]:
        """ Create the color maps.

        Parameters
        ----------
        groupsize: int
            The size of each group to display in the graph
        groups: int
            The total number of groups to graph

        Returns
        -------
        list
            The colour map for each group
        """
        colours = [] </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
        """
        If raw data isn't selected, then check the length of remaining groups until something is
        found.

        Parameters
        ----------
        raw_lines: list
            The list of keys for the raw data points
        sorted_lines:
            The list of sorted line keys to display on the graph

        Returns
        -------
        int
            The size of each group that exist within the data set.
        """
  
      @staticmethod
      def lines_groupsize(raw_lines, sorted_lines):
          """ Get the number of items in each group.
        If raw data isn't selected, then check the length of remaining groups until something is
        found.

        Parameters
        ----------
        raw_lines: list
            The list of keys for the raw data points
        sorted_lines:
            The list of sorted line keys to display on the graph

        Returns
        -------
        int
            The size of each group that exist within the data set.
        """
        If raw data isn't selected, then check the length of remaining groups until something is
        found.

        Parameters
        ----------
        raw_lines: list
            The list of keys for the raw data points
        sorted_lines:
            The list of sorted line keys to display on the graph

        Returns
        -------
        int
            The size of each group that exist within the data set.
        """
          groupsize = 1
          if raw_lines:
              groupsize = len(raw_lines)
          elif sorted_lines:
              keys = [key[0][:key[0].find("_")] for key in sorted_lines]
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