code: def data_groups(self, groups, entity_count):
          
          data = []
          # process group objects
          for xid in groups.keys():
              # get association from group data
              assoc_group_data = self.data_group_association(xid)
              data += assoc_group_data
              entity_count += len(assoc_group_data)
  
              if entity_count >= self._batch_max_chunk:
                  break
          return data, entity_count
docs: Process Group data.
  
          Args:
              groups (list): The list of groups to process.
  
          Returns:
              list: A list of groups including associations