From 95f387ac49e545b03d4a3585826de58f98d390fd Mon Sep 17 00:00:00 2001 From: anakin87 Date: Thu, 26 Sep 2024 16:28:37 +0200 Subject: [PATCH] apply suggestions --- integrations/instructor_embedders/pyproject.toml | 2 +- .../tests/test_snowflake_table_retriever.py | 14 ++++++++------ 2 files changed, 9 insertions(+), 7 deletions(-) diff --git a/integrations/instructor_embedders/pyproject.toml b/integrations/instructor_embedders/pyproject.toml index 42e71af5d..458c0ae0c 100644 --- 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 index f0fc3b99f..f5b8fee37 100644 --- 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"],