code: def readTrainingData(file_locations, GROUP_LABEL):
      '''
      Used in downstream tests
      '''
      class Mock(object):
          pass
      mock_module = Mock()
      mock_module.PARENT_LABEL = GROUP_LABEL
      for location in file_locations:
          with open(location) as f:
              tree = etree.parse(f)
  
          xml = tree.getroot()
          for each in data_prep_utils.TrainingData(xml, mock_module):
              yield eacf readTrainingData(file_locations, GROUP_LABEL):
      '''
      Used in downstream tests
      '''
      class Mock(object):
          pass
      mock_module = Mock()
      mock_module.PARENT_LABEL = GROUP_LABEL
      for location in file_locations:
          with open(location) as f:
              tree = etree.parse(f)
  
          xml = tree.getroot()
          for each in data_prep_utils.TrainingData(xml, mock_module):
              yield each
docs: Used in downstream tests