format

Author: mayinghan

tests/pytest/test_pytest_env_overwrite.py

@@ -23,8 +23,8 @@ def test_input_messages_in_decorator(row: EvaluationRow) -> EvaluationRow:
         return row
 
 
+with mock.patch.dict(os.environ, {"EP_COMPLETION_PARAMS": '[{"model": "gpt-40"}]'}):
 
-with mock.patch.dict(os.environ, {"EP_COMPLETION_PARAMS": "[{\"model\": \"gpt-40\"}]"}):
     @evaluation_test(
         input_rows=[[EvaluationRow(messages=[Message(role="user", content="What is 5 * 6?")])]],
         completion_params=[{"model": "no-op"}],  # This should be overridden by the env var
@@ -38,7 +38,6 @@ def test_input_messages_in_env(row: EvaluationRow) -> EvaluationRow:
         return row
 
 
-
 _jsonl_tmpdir = tempfile.mkdtemp()
 atexit.register(shutil.rmtree, _jsonl_tmpdir, ignore_errors=True)
 
@@ -49,6 +48,7 @@ def test_input_messages_in_env(row: EvaluationRow) -> EvaluationRow:
     )
 print(f"finish prepare input file {input_path}")
 with mock.patch.dict(os.environ, {"EP_JSONL_PATH": input_path}):
+
     @evaluation_test(
         input_rows=[[EvaluationRow(messages=[Message(role="user", content="This will be ignored")])]],
         completion_params=[{"model": "no-op"}],