如何用luigi将输出写入orc格式的分区表？

import luigi
import pyorc
import os

class ORCPartitionTask(luigi.Task):
    input_file = luigi.Parameter()
    output_dir = luigi.Parameter()

    def output(self):
        return luigi.LocalTarget(self.output_dir)

    def run(self):
        # 在这里编写将输出写入ORC格式的分区表的代码
        # 可以使用pyorc库来读取和写入ORC文件
        # 可以使用os库来创建输出目录和分区目录
        pass

def run(self):
    with open(self.input_file, 'rb') as f:
        reader = pyorc.Reader(f)
        schema = reader.schema

        # 获取ORC文件中的分区列
        partition_columns = schema.get_partition_columns()

        # 创建输出目录
        os.makedirs(self.output_dir, exist_ok=True)

        # 遍历分区列的所有可能取值
        for partition_value in partition_columns:
            # 创建分区目录
            partition_dir = os.path.join(self.output_dir, partition_value)
            os.makedirs(partition_dir, exist_ok=True)

            # 构建输出文件路径
            output_file = os.path.join(partition_dir, 'data.orc')

            # 创建ORC写入器
            writer = pyorc.Writer(output_file, schema)

            # 遍历ORC文件中的每一行数据
            for row in reader:
                # 在这里根据分区列的值将数据写入相应的分区文件
                # 可以使用writer.writerow方法来写入数据

            # 关闭ORC写入器
            writer.close()

if __name__ == '__main__':
    luigi.build([ORCPartitionTask(input_file='input.orc', output_dir='output')], local_scheduler=True)