diff --git a/src/pytorch_ie/metrics/statistics.py b/src/pytorch_ie/metrics/statistics.py index c8a5ce4a..05d474c0 100644 --- a/src/pytorch_ie/metrics/statistics.py +++ b/src/pytorch_ie/metrics/statistics.py @@ -132,7 +132,9 @@ def __init__( f"functions that rely on zero values." ) self.aggregation_functions = { - name: func for name, func in self.aggregation_functions.items() if name not in ["mean", "std", "min"] + name: func + for name, func in self.aggregation_functions.items() + if name not in ["mean", "std", "min"] } self.labels = labels