refactor(pdf_parse_union_core_v2): update import paths to use new pac…

…kage structure Adapt import statements in `pdf_parse_union_core_v2.py` to reflect the updated packagestructure, changing from the `magic_pdf.v3.helpers` module to the `magic_pdf.model.v3` module. This ensures compatibility with the revised directory layout.
opendatalab · Sep 28, 2024 · 5522d0a · 5522d0a
1 parent 2145a8b
commit 5522d0a
Show file tree

Hide file tree

Showing 7 changed files with 1 addition and 1 deletion.
diff --git a/magic_pdf/model/v3/__init__.py b/magic_pdf/model/v3/__init__.py
diff --git a/magic_pdf/v3/ds_config.json → magic_pdf/model/v3/ds_config.json b/magic_pdf/v3/ds_config.json → magic_pdf/model/v3/ds_config.json
diff --git a/magic_pdf/v3/eval.py → magic_pdf/model/v3/eval.py b/magic_pdf/v3/eval.py → magic_pdf/model/v3/eval.py
diff --git a/magic_pdf/v3/helpers.py → magic_pdf/model/v3/helpers.py b/magic_pdf/v3/helpers.py → magic_pdf/model/v3/helpers.py
diff --git a/magic_pdf/v3/train.py → magic_pdf/model/v3/train.py b/magic_pdf/v3/train.py → magic_pdf/model/v3/train.py
diff --git a/magic_pdf/v3/train.sh → magic_pdf/model/v3/train.sh b/magic_pdf/v3/train.sh → magic_pdf/model/v3/train.sh
diff --git a/magic_pdf/pdf_parse_union_core_v2.py b/magic_pdf/pdf_parse_union_core_v2.py
@@ -140,7 +140,7 @@ def get_model(self, model_name: str, local_path=None):
 
 
 def do_predict(boxes: List[List[int]], model) -> List[int]:
-    from magic_pdf.v3.helpers import prepare_inputs, boxes2inputs, parse_logits
+    from magic_pdf.model.v3 import prepare_inputs, boxes2inputs, parse_logits
     inputs = boxes2inputs(boxes)
     inputs = prepare_inputs(inputs, model)
     logits = model(**inputs).logits.cpu().squeeze(0)