Skip to content

Commit

Permalink
log a warning and remove "mean", "std", "min" from aggregation_functi…
Browse files Browse the repository at this point in the history
…ons if labels are inferred from data
  • Loading branch information
ArneBinder committed Sep 27, 2023
1 parent e9f6829 commit c292b31
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions src/pytorch_ie/metrics/statistics.py
Original file line number Diff line number Diff line change
Expand Up @@ -131,9 +131,9 @@ def __init__(
f"this collector, but be aware that the results may be wrong for your own aggregation "
f"functions that rely on zero values."
)
self.aggregation_functions = [
func for func in self.aggregation_functions if func not in ["mean", "std", "min"]
]
self.aggregation_functions = {
name: func for name, func in self.aggregation_functions.items() if name not in ["mean", "std", "min"]
}

self.labels = labels

Expand Down

0 comments on commit c292b31

Please sign in to comment.