From 850b57b9cb3e7ec2e744cb3e92cfd57ce39f9de5 Mon Sep 17 00:00:00 2001 From: Silvano Cerza Date: Wed, 10 Jan 2024 09:57:08 +0100 Subject: [PATCH] Fix linting --- haystack/evaluation/eval.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/haystack/evaluation/eval.py b/haystack/evaluation/eval.py index 6fdc7138a6..28b4b76bbe 100644 --- a/haystack/evaluation/eval.py +++ b/haystack/evaluation/eval.py @@ -40,7 +40,6 @@ def __init__( Metric.SAS: self._calculate_sas, } - # pylint: disable=too-many-return-statements def calculate_metrics(self, metric: Union[Metric, Callable[..., MetricsResult]], **kwargs) -> MetricsResult: """ Calculate evaluation metrics based on the provided Metric or using the custom metric function. @@ -49,7 +48,7 @@ def calculate_metrics(self, metric: Union[Metric, Callable[..., MetricsResult]], :return: MetricsResult containing the calculated metric. """ - if metric in self._supported_metrics: + if isinstance(metric, Metric): return self._supported_metrics[metric](**kwargs) return metric(self, **kwargs)