Skip to content

Commit 73071dc

Browse files
authored
Hotfix/name mismatch (#40387)
1 parent ced749c commit 73071dc

File tree

1 file changed

+6
-0
lines changed
  • sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/red_team

1 file changed

+6
-0
lines changed

sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/red_team/_red_team.py

+6
Original file line numberDiff line numberDiff line change
@@ -264,6 +264,10 @@ async def _log_redteam_results_to_mlflow(
264264
if data_only:
265265
f.write(json.dumps({"conversations": redteam_output.attack_details or []}))
266266
elif redteam_output.scan_result:
267+
redteam_output.scan_result["redteaming_scorecard"] = redteam_output.scan_result.get("scorecard", None)
268+
redteam_output.scan_result["redteaming_parameters"] = redteam_output.scan_result.get("parameters", None)
269+
redteam_output.scan_result["redteaming_data"] = redteam_output.scan_result.get("attack_details", None)
270+
267271
json.dump(redteam_output.scan_result, f)
268272

269273
# Copy all relevant files to the temp directory
@@ -276,6 +280,8 @@ async def _log_redteam_results_to_mlflow(
276280
continue
277281
if file.endswith('.log') and not os.environ.get('DEBUG'):
278282
continue
283+
if file == artifact_name or file == eval_info_name:
284+
continue
279285

280286
try:
281287
shutil.copy(file_path, os.path.join(tmpdir, file))

0 commit comments

Comments
 (0)