如何使用apache beam将列表对象写入JSON文件？

import apache_beam as beam
import json

class WriteToJson(beam.DoFn):
    def __init__(self, output_file):
        self.output_file = output_file

    def process(self, element):
        with open(self.output_file, 'a') as f:
            json.dump(element, f)
            f.write('\n')

with beam.Pipeline() as pipeline:
    data = pipeline | beam.Create([{'name': 'Alice', 'age': 25}, {'name': 'Bob', 'age': 30}])

    data | beam.ParDo(WriteToJson('output.json'))