updating linting

deepset-ai · Nov 14, 2024 · adbfef5 · adbfef5
1 parent cd2658f
commit adbfef5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/...ke/src/haystack_integrations/components/retrievers/snowflake/snowflake_table_retriever.py b/...ke/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
@@ -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,
             },
         }