fix pytest

triton-inference-server · Sep 6, 2024 · e8bfc4c · e8bfc4c
1 parent ba9d5f1
commit e8bfc4c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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):