apply suggestions

deepset-ai · Sep 26, 2024 · 95f387a · 95f387a
1 parent 4f497c6
commit 95f387a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/integrations/instructor_embedders/pyproject.toml b/integrations/instructor_embedders/pyproject.toml
@@ -155,7 +155,7 @@ unfixable = [
 ]
 
 [tool.ruff.lint.isort]
-known-first-party = ["instructor_embedders"]
+known-first-party = ["haystack_integrations"]
 
 [tool.ruff.lint.flake8-tidy-imports]
 ban-relative-imports = "parents"

diff --git a/integrations/snowflake/tests/test_snowflake_table_retriever.py b/integrations/snowflake/tests/test_snowflake_table_retriever.py
@@ -477,10 +477,11 @@ def test_to_dict_default(self, monkeypatch: MagicMock) -> None:
 
         data = component.to_dict()
 
-        t = "haystack_integrations.components.retrievers.snowflake.snowflake_table_retriever.SnowflakeTableRetriever"
-
         assert data == {
-            "type": t,
+            "type": (
+                "haystack_integrations.components.retrievers.snowflake."
+                "snowflake_table_retriever.SnowflakeTableRetriever"
+            ),
             "init_parameters": {
                 "api_key": {
                     "env_vars": ["SNOWFLAKE_API_KEY"],
@@ -511,10 +512,11 @@ def test_to_dict_with_parameters(self, monkeypatch: MagicMock) -> None:
 
         data = component.to_dict()
 
-        t = "haystack_integrations.components.retrievers.snowflake.snowflake_table_retriever.SnowflakeTableRetriever"
-
         assert data == {
-            "type": t,
+            "type": (
+                "haystack_integrations.components.retrievers.snowflake."
+                "snowflake_table_retriever.SnowflakeTableRetriever"
+            ),
             "init_parameters": {
                 "api_key": {
                     "env_vars": ["SNOWFLAKE_API_KEY"],