Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 15 additions & 1 deletion src/lighteval/tasks/lighteval_task.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@

from lighteval.metrics.metrics import Metrics
from lighteval.metrics.metrics_sample import SamplingMetric
from lighteval.metrics.utils.metric_utils import Metric
from lighteval.metrics.utils.metric_utils import Metric, MetricGrouping
from lighteval.tasks.prompt_manager import FewShotSampler
from lighteval.tasks.requests import (
Doc,
Expand Down Expand Up @@ -167,7 +167,21 @@ def __str__(self, lite: bool = False):
continue
if k == "metrics":
for ix, metrics in enumerate(v):
is_metric_grouping = False
if isinstance(getattr(self, k)[ix], MetricGrouping):
is_metric_grouping = True
for metric_k, metric_v in metrics.items():
if is_metric_grouping and isinstance(metric_v, dict):
for metric_sub_k, metric_sub_v in metric_v.items():
if isinstance(metric_sub_v, Callable):
repr_v = metric_sub_v.__name__
elif isinstance(metric_sub_v, Metric.get_allowed_types_for_metrics()):
repr_v = str(metric_sub_v)
else:
repr_v = repr(metric_sub_v)
values.append([f"{k} {ix}: {metric_k}: {metric_sub_k}", repr_v])
continue
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

you can remove the continue here and use elif right below


if isinstance(metric_v, Callable):
repr_v = metric_v.__name__
elif isinstance(metric_v, Metric.get_allowed_types_for_metrics()):
Expand Down
Loading