Skip to content

Commit a7d7b11

Browse files
authored
Rename PDF merger path keys in pipeline
1 parent d279e94 commit a7d7b11

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

dataflow/statics/pipelines/api_pipelines/pdf_vqa_extract_pipeline.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -42,9 +42,9 @@ def __init__(self):
4242
def forward(self):
4343
self.pdf_merger.run(
4444
storage=self.storage.step(),
45-
input_pdf_path_key="input_pdf_paths",
45+
input_pdf_list_key="input_pdf_paths",
4646
input_name_key="name",
47-
output_merged_pdf_path_key="merged_pdf_path",
47+
output_pdf_path_key="merged_pdf_path",
4848
)
4949
self.mineru_executor.run(
5050
storage=self.storage.step(),
@@ -83,4 +83,4 @@ def forward(self):
8383
# jsonl中每一行包含input_pdf_path, name (math1, math2, physics1, chemistry1, ...)
8484
pipeline = PDF_VQA_extract_optimized_pipeline()
8585
pipeline.compile()
86-
pipeline.forward()
86+
pipeline.forward()

0 commit comments

Comments
 (0)