Skip to content

Commit 60e7480

Browse files
committed
fix: fix ruff errors
1 parent c8b8c37 commit 60e7480

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

src/pruna/evaluation/metrics/metric_evalharness.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@
1616

1717
from typing import Any, List, Tuple
1818

19-
import numpy as np
2019
from lm_eval.api import metrics # noqa: F401 # needed to register lm-eval metrics
2120
from lm_eval.api import registry as lm_registry
2221

src/pruna/evaluation/task.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -243,7 +243,7 @@ def _process_metric_names(
243243
metric_name = cast(str, metric_name)
244244
new_requests.append(cast(str, metric_name))
245245
return MetricRegistry.get_metrics(
246-
names=new_requests, inference_device=inference_device, stateful_metric_device=stateful_metric_device
246+
names=new_requests, inference_device=inference_device, stateful_metric_device=stateful_metric_device
247247
)
248248

249249

0 commit comments

Comments
 (0)