From 6c3d31d68b55d35ee3c65c893b5433dc06fbb8c6 Mon Sep 17 00:00:00 2001 From: Michele Dolfi Date: Fri, 24 Jan 2025 16:00:22 +0100 Subject: [PATCH] fix artifacts_path type Signed-off-by: Michele Dolfi --- docling/models/code_formula_model.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/docling/models/code_formula_model.py b/docling/models/code_formula_model.py index 5c23dcc8..e9c5e52b 100644 --- a/docling/models/code_formula_model.py +++ b/docling/models/code_formula_model.py @@ -1,6 +1,6 @@ import re from pathlib import Path -from typing import Iterable, List, Literal, Optional, Tuple +from typing import Iterable, List, Literal, Optional, Tuple, Union from docling_core.types.doc import CodeItem, DoclingDocument, NodeItem, TextItem from docling_core.types.doc.base import BoundingBox @@ -62,7 +62,7 @@ class CodeFormulaModel(BaseItemAndImageEnrichmentModel): def __init__( self, enabled: bool, - artifacts_path: Optional[Path], + artifacts_path: Optional[Union[Path, str]], options: CodeFormulaModelOptions, accelerator_options: AcceleratorOptions, ): @@ -92,6 +92,8 @@ class CodeFormulaModel(BaseItemAndImageEnrichmentModel): if artifacts_path is None: artifacts_path = self.download_models_hf() + else: + artifacts_path = Path(artifacts_path) self.code_formula_model = CodeFormulaPredictor( artifacts_path=artifacts_path,