Skip to content
Merged
Prev Previous commit
Next Next commit
Update test_docling_utils.py
  • Loading branch information
erichare committed Dec 2, 2025
commit a86364e57fee8c7fc80fbda912bc76bb968498de
9 changes: 8 additions & 1 deletion src/lfx/tests/unit/base/data/test_docling_utils.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,19 @@
"""Tests for docling_utils module."""

import pytest
from docling_core.types.doc import DoclingDocument

try:
from docling_core.types.doc import DoclingDocument
DOCLING_AVAILABLE = True
except ImportError:
DOCLING_AVAILABLE = False

from lfx.base.data.docling_utils import extract_docling_documents
from lfx.schema.data import Data
from lfx.schema.dataframe import DataFrame


@pytest.mark.skipif(not DOCLING_AVAILABLE, reason="docling_core not installed")
class TestExtractDoclingDocuments:
"""Test extract_docling_documents function."""

Expand Down
Loading