Skip to content

Commit

Permalink
chore: fix test linting problems
Browse files Browse the repository at this point in the history
  • Loading branch information
FHardow committed Aug 12, 2024
1 parent ccdde1f commit c1d0a19
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions integrations/opensearch/tests/test_auth.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
from unittest.mock import Mock, patch

import pytest
from haystack.components.retrievers import SentenceWindowRetrieval
from haystack_integrations.document_stores.opensearch.auth import AWSAuth
from opensearchpy import Urllib3AWSV4SignerAuth
from haystack.components.retrievers import SentenceWindowRetrieval


class TestAWSAuth:
Expand Down Expand Up @@ -114,15 +114,15 @@ def test_call(self, _get_urllib3_aws_v4_signer_auth_mock):
signer_auth_mock.assert_called_once_with("GET", "http://some.url", "some body")

@patch("haystack_integrations.document_stores.opensearch.auth.AWSAuth._get_urllib3_aws_v4_signer_auth")
def test_sentence_window_retriever(self, _get_urllib3_aws_v4_signer_auth_mock, mock_boto3_session):
def test_sentence_window_retriever(self, _get_urllib3_aws_v4_signer_auth_mock):
data = {
"init_parameters": {
"document_store": {
"init_parameters": {
"embedding_dim": 1024,
"hosts": ["${OPENSEARCH_HOST}:${OPENSEARCH_PORT}"],
"http_auth": {"type": "haystack_integrations.document_stores.opensearch.auth.AWSAuth"},
"index": "0a4680f5-f96c-4315-96a0-1c61373077ba-9b24e10f-d5b8-45d4-b844-d848ee3d3fa6-594cdce3-05eb-472a-9ab7-3493242ad463",
"index": "random-test-index",
"similarity": "cosine",
"use_ssl": True,
"verify_certs": False,
Expand Down

0 comments on commit c1d0a19

Please sign in to comment.