Fixed liniting.

langchain-ai · May 8, 2024 · 6b49042 · 6b49042
1 parent fe7ad44
commit 6b49042
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/libs/aws/tests/unit_tests/retrievers/test_bedrock.py b/libs/aws/tests/unit_tests/retrievers/test_bedrock.py
@@ -1,9 +1,12 @@
+# type: ignore
+
 from unittest.mock import MagicMock
 
 import pytest
-from langchain_community.retrievers import AmazonKnowledgeBasesRetriever
 from langchain_core.documents import Document
 
+from langchain_aws.retrievers import AmazonKnowledgeBasesRetriever
+
 
 @pytest.fixture
 def mock_client():