From 0495d11809a0695adc380ca1d1e023bf1a2d8090 Mon Sep 17 00:00:00 2001 From: Erick Galinkin Date: Thu, 8 Feb 2024 09:30:36 -0500 Subject: [PATCH] Avoid divide by zero error (#458) --- garak/report.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/garak/report.py b/garak/report.py index 3a308557..6a4197cf 100644 --- a/garak/report.py +++ b/garak/report.py @@ -71,7 +71,7 @@ def get_evaluations(self): evals_df = pd.DataFrame.from_dict(evals) self.evaluations = evals_df.assign( - score=lambda x: (x["passed"] / x["total"] * 100) + score=lambda x: (x["passed"] / x["total"] * 100) if x["total"] > 0 else 0 ) self.scores = self.evaluations[["probe", "score"]].groupby("probe").mean() return self