Skip to content

Commit fcaff6e

Browse files
nagkumar91Nagkumar ArkalgudNagkumar Arkalgud
authored
Add reason to end run for red team (Azure#40777)
* aggregated binary threshold results for evaluators in metrics * handle when the right columns do not exist * Marking mypy, pylint, black as false * Add to constatns * typo * End with reason --------- Co-authored-by: Nagkumar Arkalgud <[email protected]> Co-authored-by: Nagkumar Arkalgud <[email protected]>
1 parent 19c8b26 commit fcaff6e

File tree

1 file changed

+1
-1
lines changed
  • sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/red_team

1 file changed

+1
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -462,7 +462,7 @@ async def _log_redteam_results_to_mlflow(
462462
if key != "risk_category":
463463
eval_run.log_metric(f"{risk_category}_{key}", cast(float, value))
464464
self.logger.debug(f"Logged metric: {risk_category}_{key} = {value}")
465-
eval_run._end_run()
465+
eval_run._end_run("FINISHED")
466466
self.logger.info("Successfully logged results to MLFlow")
467467
return None
468468

0 commit comments

Comments
 (0)