Skip to content

Commit 705e014

Browse files
Update GANDLF/metrics/generic.py
Co-authored-by: Sarthak Pati <patis@iu.edu>
1 parent 429b59e commit 705e014

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

GANDLF/metrics/generic.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -15,13 +15,7 @@ def define_average_type_key(params, metric_name):
1515
"""Determine if the metric config defines the type of average to use.
1616
If not, fallback to the default (macro) average type.
1717
"""
18-
if "average" in params["metrics"][metric_name]:
19-
average_type_key = params["metrics"][metric_name]["average"]
20-
else:
21-
average_type_key = "macro"
22-
UserWarning(
23-
"WARNING: Average type not defined in config, using default (macro)."
24-
)
18+
average_type_key = params["metrics"][metric_name].get("average", "macro")
2519
return average_type_key
2620

2721

0 commit comments

Comments
 (0)