def METHOD_NAME(self, dst_partition_id):
    res = []
    for offset, output_val in enumerate(self._output_vals):
        outputs = output_val.get()
        for val_pos in outputs:
            if val_pos.partition_id == dst_partition_id:
                res.append(offset)
    return res