diff --git a/integrations/snowflake/src/haystack_integrations/components/retrievers/snowflake/snowflake_table_retriever.py b/integrations/snowflake/src/haystack_integrations/components/retrievers/snowflake/snowflake_table_retriever.py index 09a575ced..d2b7744fb 100644 --- a/integrations/snowflake/src/haystack_integrations/components/retrievers/snowflake/snowflake_table_retriever.py +++ b/integrations/snowflake/src/haystack_integrations/components/retrievers/snowflake/snowflake_table_retriever.py @@ -299,9 +299,7 @@ def _fetch_data( if self.private_key_file is None: connect_params["password"] = self.api_key.resolve_value() # Create a new connection with every run - conn = self._snowflake_connector( - connect_params=connect_params - ) + conn = self._snowflake_connector(connect_params=connect_params) if conn is None: return df except (ForbiddenError, ProgrammingError) as e: diff --git a/integrations/snowflake/tests/test_snowflake_table_retriever.py b/integrations/snowflake/tests/test_snowflake_table_retriever.py index 4aa990a6c..5991ba9da 100644 --- a/integrations/snowflake/tests/test_snowflake_table_retriever.py +++ b/integrations/snowflake/tests/test_snowflake_table_retriever.py @@ -474,7 +474,7 @@ def test_to_dict_default(self, monkeypatch: MagicMock) -> None: warehouse="test_warehouse", login_timeout=30, private_key_file=None, - private_key_file_pwd=None + private_key_file_pwd=None, ) data = component.to_dict() @@ -497,7 +497,7 @@ def test_to_dict_default(self, monkeypatch: MagicMock) -> None: "warehouse": "test_warehouse", "login_timeout": 30, "private_key_file": None, - "private_key_file_pwd": None + "private_key_file_pwd": None, }, } @@ -534,7 +534,7 @@ def test_to_dict_with_parameters(self, monkeypatch: MagicMock) -> None: "warehouse": "COMPUTE_WH", "login_timeout": 30, "private_key_file": None, - "private_key_file_pwd": None + "private_key_file_pwd": None, }, }