Fix failing test

microsoft · Apr 18, 2024 · 3197fed · 3197fed
1 parent d56b607
commit 3197fed
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/src/promptflow-evals/tests/evals/unittests/test_simulator.py b/src/promptflow-evals/tests/evals/unittests/test_simulator.py
@@ -19,14 +19,14 @@
 
 @pytest.fixture()
 def mock_config():
-    mock_config = Mock()
-    mock_config.api_key = "apikey"
-    mock_config.deployment_name = "deployment"
-    mock_config.api_version = "api-version"
-    mock_config.api_base = "api-base"
-    mock_config.model_name = "model-name"
-    mock_config.model_kwargs = {}
-    yield mock_config
+    return {
+        "api_key": "apikey",
+        "deployment_name": "deployment",
+        "api_version": "api-version",
+        "api_base": "api-base",
+        "model_name": "model-name",
+        "model_kwargs": {},
+    }
 
 
 @pytest.fixture()