From 0f48e6c36eb23a5c5fdcd3c15ec31fa0c4dfd5f5 Mon Sep 17 00:00:00 2001
From: Bagatur <22008038+baskaryan@users.noreply.github.com>
Date: Thu, 24 Aug 2023 15:06:53 -0700
Subject: [PATCH] fix integration deps (#9722)

---
 .../integration_tests/document_loaders/test_polars_dataframe.py | 2 ++
 .../tests/integration_tests/vectorstores/test_elasticsearch.py  | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

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(),