code: def partition(self, mapped_values):
          partitioned_data = collections.defaultdict(list)
          for key, value in mapped_values:
              partitioned_data[key].append(value)
          return partitioned_data.items()