From 60a83659d5382152421987c014c575e4203b502d Mon Sep 17 00:00:00 2001 From: Adam Dougal Date: Wed, 8 May 2024 16:32:43 +0000 Subject: [PATCH] Yield llmhelper --- .../processors/test_IntegratedVectorizationEmbedder.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/code/tests/utilities/helpers/processors/test_IntegratedVectorizationEmbedder.py b/code/tests/utilities/helpers/processors/test_IntegratedVectorizationEmbedder.py index 76d6cc6e6..dc1b8b0ec 100644 --- a/code/tests/utilities/helpers/processors/test_IntegratedVectorizationEmbedder.py +++ b/code/tests/utilities/helpers/processors/test_IntegratedVectorizationEmbedder.py @@ -33,7 +33,7 @@ def llm_helper_mock(): 0 ] * 1536 llm_helper.generate_embeddings.return_value = [123] - yield mock + yield llm_helper @pytest.fixture(autouse=True) @@ -99,7 +99,7 @@ def test_process_using_integrated_vectorization( azure_search_iv_datasource_helper_mock.return_value.create_or_update_datasource.assert_called_once() azure_search_iv_index_helper_mock.assert_called_once_with( - env_helper_mock, llm_helper_mock.return_value + env_helper_mock, llm_helper_mock ) azure_search_iv_index_helper_mock.return_value.create_or_update_index.assert_called_once()