Skip to content

Commit 270ca43

Browse files
authored
fix: division by 0 (#105)
1 parent 3f0e268 commit 270ca43

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/mlipaudit/scoring.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ def compute_model_score(scores: dict[str, float | None]) -> float:
9696
ValueError: If 'Overall score' is a key in the scores dictionary.
9797
9898
Returns:
99-
The mean score.
99+
The mean score if scores are available. Otherwise returns 0.
100100
"""
101101
if "Overall score" in scores:
102102
raise ValueError("Overall score should not be part of dictionary.")
@@ -108,4 +108,7 @@ def compute_model_score(scores: dict[str, float | None]) -> float:
108108
name: score for name, score in scores.items() if score is not None
109109
}
110110

111-
return sum(benchmarks_with_scores.values()) / len(benchmarks_with_scores)
111+
if benchmarks_with_scores:
112+
return sum(benchmarks_with_scores.values()) / len(benchmarks_with_scores)
113+
114+
return 0.0

0 commit comments

Comments
 (0)