@@ -525,9 +525,7 @@ async def test_resolve_provider_variables_empty_provider_vars(self, component_cl
525525
526526 assert result == {}
527527
528- async def test_resolve_provider_variables_variable_service_returns_none (
529- self , component_class , default_kwargs
530- ):
528+ async def test_resolve_provider_variables_variable_service_returns_none (self , component_class , default_kwargs ):
531529 """Test _resolve_provider_variables when variable_service is None."""
532530 component = component_class (** default_kwargs )
533531
@@ -784,9 +782,7 @@ def test_build_embeddings_watsonx_api_key_from_provider_vars(
784782 assert result == mock_embeddings
785783
786784 @patch ("langchain_ibm.WatsonxEmbeddings" )
787- def test_build_embeddings_watsonx_partial_vars (
788- self , mock_watsonx_embeddings , component_class , default_kwargs
789- ):
785+ def test_build_embeddings_watsonx_partial_vars (self , mock_watsonx_embeddings , component_class , default_kwargs ):
790786 """Test WatsonX with only apikey, no project_id or url."""
791787 component = component_class (** default_kwargs )
792788
@@ -896,9 +892,7 @@ async def test_retrieve_data_with_search_query(self, component_class, default_kw
896892 result = await component .retrieve_data ()
897893
898894 # Verify similarity_search_with_score was used (not similarity_search)
899- mock_chroma_instance .similarity_search_with_score .assert_called_once_with (
900- query = "find me something" , k = 5
901- )
895+ mock_chroma_instance .similarity_search_with_score .assert_called_once_with (query = "find me something" , k = 5 )
902896 mock_chroma_instance .similarity_search .assert_not_called ()
903897 assert len (result ) == 1
904898
0 commit comments