diff --git a/test_unstructured/documents/test_elements.py b/test_unstructured/documents/test_elements.py index 1a7176d68b..c79c7b08da 100644 --- a/test_unstructured/documents/test_elements.py +++ b/test_unstructured/documents/test_elements.py @@ -18,7 +18,6 @@ RelativeCoordinateSystem, ) from unstructured.documents.elements import ( - UUID, ConsolidationStrategy, CoordinatesMetadata, DataSourceMetadata, diff --git a/test_unstructured/documents/test_email_elements.py b/test_unstructured/documents/test_email_elements.py index b2de8e542e..865fdf8b61 100644 --- a/test_unstructured/documents/test_email_elements.py +++ b/test_unstructured/documents/test_email_elements.py @@ -1,11 +1,10 @@ -import uuid from functools import partial import pytest from unstructured.cleaners.core import clean_prefix from unstructured.cleaners.translate import translate_text -from unstructured.documents.email_elements import UUID, EmailElement, Name, NoID +from unstructured.documents.email_elements import EmailElement, Name, NoID def test_text_id(): diff --git a/test_unstructured/partition/docx/test_doc.py b/test_unstructured/partition/docx/test_doc.py index 993b16e203..bb2204f08d 100644 --- a/test_unstructured/partition/docx/test_doc.py +++ b/test_unstructured/partition/docx/test_doc.py @@ -17,7 +17,6 @@ from unstructured.partition.common import convert_office_doc from unstructured.partition.doc import partition_doc from unstructured.partition.docx import partition_docx -from unstructured.staging.base import convert_to_dataframe @pytest.fixture() diff --git a/test_unstructured/partition/pdf_image/test_pdf.py b/test_unstructured/partition/pdf_image/test_pdf.py index 93cf805c0a..5c87dd0288 100644 --- a/test_unstructured/partition/pdf_image/test_pdf.py +++ b/test_unstructured/partition/pdf_image/test_pdf.py @@ -1234,7 +1234,7 @@ def test_partition_pdf_always_keep_all_image_elements( assert len(image_elements) == 3 -@pytest.fixture +@pytest.fixture() def mock_pdfwriter_with_duplicate_pages(): test_file = Path("example-docs") / "DA-1p.pdf" original_pdf = PdfReader(test_file) @@ -1248,7 +1248,7 @@ def mock_pdfwriter_with_duplicate_pages(): return writer -@pytest.fixture +@pytest.fixture() def expected_element_ids_for_fast_strategy(): return [ "a2744e3d8b632d388a1f24c8806f1a6e", @@ -1282,7 +1282,7 @@ def expected_element_ids_for_fast_strategy(): ] -@pytest.fixture +@pytest.fixture() def expected_element_ids_for_hi_res_strategy(): return [ "a2744e3d8b632d388a1f24c8806f1a6e", @@ -1316,7 +1316,7 @@ def expected_element_ids_for_hi_res_strategy(): ] -@pytest.fixture +@pytest.fixture() def expected_element_ids_for_ocr_strategy(): return [ "a2744e3d8b632d388a1f24c8806f1a6e", @@ -1348,7 +1348,7 @@ def expected_element_ids_for_ocr_strategy(): ] -@pytest.fixture +@pytest.fixture() def expected_ids(request): return request.getfixturevalue(request.param) diff --git a/unstructured/documents/email_elements.py b/unstructured/documents/email_elements.py index 3cd359c260..7fd63de963 100644 --- a/unstructured/documents/email_elements.py +++ b/unstructured/documents/email_elements.py @@ -4,7 +4,7 @@ from datetime import datetime from typing import Callable, List, Union -from unstructured.documents.elements import UUID, Element, NoID, Text +from unstructured.documents.elements import Element, NoID, Text class NoDatestamp(ABC):