From e8bfc4c96549832406be0ec56ca0ac7e60269bc8 Mon Sep 17 00:00:00 2001 From: Hyunjae Woo Date: Fri, 6 Sep 2024 09:53:19 -0700 Subject: [PATCH] fix pytest --- genai-perf/tests/test_synthetic_data_retriever.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/genai-perf/tests/test_synthetic_data_retriever.py b/genai-perf/tests/test_synthetic_data_retriever.py index 5932663c..a1959b73 100644 --- a/genai-perf/tests/test_synthetic_data_retriever.py +++ b/genai-perf/tests/test_synthetic_data_retriever.py @@ -17,13 +17,13 @@ import pytest from genai_perf.inputs.input_constants import OutputFormat from genai_perf.inputs.inputs_config import InputsConfig -from genai_perf.inputs.synthetic_retriever import SyntheticDataRetriever +from genai_perf.inputs.synthetic_data_retriever import SyntheticDataRetriever class TestSyntheticDataRetriever: @patch( - "genai_perf.inputs.synthetic_retriever.SyntheticPromptGenerator.create_synthetic_prompt", + "genai_perf.inputs.synthetic_data_retriever.SyntheticPromptGenerator.create_synthetic_prompt", return_value="test prompt", ) @pytest.mark.parametrize( @@ -51,11 +51,11 @@ def test_synthetic_text(self, mock_prompt, output_format): ] @patch( - "genai_perf.inputs.synthetic_retriever.SyntheticPromptGenerator.create_synthetic_prompt", + "genai_perf.inputs.synthetic_data_retriever.SyntheticPromptGenerator.create_synthetic_prompt", return_value="test prompt", ) @patch( - "genai_perf.inputs.synthetic_retriever.SyntheticImageGenerator.create_synthetic_image", + "genai_perf.inputs.synthetic_data_retriever.SyntheticImageGenerator.create_synthetic_image", return_value="data:image/jpeg;base64,test_base64_encoding", ) def test_synthetic_text_and_image(self, mock_prompt, mock_image):