Skip to content

GitLab

Commit 91f7cff8 authored by 赵小蒙's avatar 赵小蒙
Browse files

fix replace_equations without "score" bug

parent e5907296
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
magic_pdf/pdf_parse_union_core.py
View file @ 91f7cff8
......@@ -65,6 +65,7 @@ def txt_spans_extract(pdf_page, inline_equations, interline_equations):
"bbox": list(span["bbox"]),
"content": span["latex"],
"type": ContentType.InlineEquation,
"score": 1.0,
}
)
elif span.get('type') == ContentType.InterlineEquation:
......@@ -73,6 +74,7 @@ def txt_spans_extract(pdf_page, inline_equations, interline_equations):
"bbox": list(span["bbox"]),
"content": span["latex"],
"type": ContentType.InterlineEquation,
"score": 1.0,
}
)
else:
......@@ -81,6 +83,7 @@ def txt_spans_extract(pdf_page, inline_equations, interline_equations):
"bbox": list(span["bbox"]),
"content": span["text"],
"type": ContentType.Text,
"score": 1.0,
}
)
return spans
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment