diff --git a/libs/langchain/tests/integration_tests/document_loaders/test_polars_dataframe.py b/libs/langchain/tests/integration_tests/document_loaders/test_polars_dataframe.py index b9e4727d20b..03f5070120b 100644 --- a/libs/langchain/tests/integration_tests/document_loaders/test_polars_dataframe.py +++ b/libs/langchain/tests/integration_tests/document_loaders/test_polars_dataframe.py @@ -1,3 +1,5 @@ +from __future__ import annotations + from typing import TYPE_CHECKING import pytest diff --git a/libs/langchain/tests/integration_tests/vectorstores/test_elasticsearch.py b/libs/langchain/tests/integration_tests/vectorstores/test_elasticsearch.py index f2e451ee718..dafba4b1f79 100644 --- a/libs/langchain/tests/integration_tests/vectorstores/test_elasticsearch.py +++ b/libs/langchain/tests/integration_tests/vectorstores/test_elasticsearch.py @@ -5,7 +5,6 @@ import uuid from typing import Generator, List, Union import pytest -from elasticsearch.helpers import BulkIndexError from langchain.docstore.document import Document from langchain.vectorstores.elasticsearch import ElasticsearchStore @@ -585,6 +584,7 @@ class TestElasticsearch: caplog: pytest.LogCaptureFixture, ) -> None: """Test bulk exception logging is giving better hints.""" + from elasticsearch.helpers import BulkIndexError docsearch = ElasticsearchStore( embedding=ConsistentFakeEmbeddings(),