在apache beam中将字符串列转换为整数列？

import apache_beam as beam

class StringToInt(beam.DoFn):
    def process(self, element):
        # 将字符串转换为整数
        integer_value = int(element)
        # 返回转换后的整数
        yield integer_value

# 创建一个 Pipeline 对象
p = beam.Pipeline()

# 从输入文件读取数据
input_data = p | 'Read from file' >> beam.io.ReadFromText('input.txt')

# 使用 ParDo 函数将字符串列转换为整数列
output_data = input_data | 'Convert to integer' >> beam.ParDo(StringToInt())

# 将转换后的整数列写入输出文件
output_data | 'Write to file' >> beam.io.WriteToText('output.txt')

# 运行 Pipeline
p.run()