changes for pre-commit

lm_eval/filters/selection.py

@@ -15,8 +15,8 @@ class TakeFirstFilter(Filter):
         """
         return map(lambda r: r[0], resps)
 
-class TakeKFilter(Filter):
 
+class TakeKFilter(Filter):
     def __init__(self, *args, **kwargs):
 
         self.k = kwargs.pop("k")
@@ -25,8 +25,10 @@ class TakeKFilter(Filter):
 
     def apply(self, resps):
         # check we have at least k responses per doc, else we can't take the first k
-        assert len(resps[0]) >= self.k, f"Need at least {self.k} responses per doc to take first {self.k}, but got {len(resps[0])} only! Please increase TaskConfig.repeats ."
-        return map(lambda r: r[:self.k], resps)
+        assert (
+            len(resps[0]) >= self.k
+        ), f"Need at least {self.k} responses per doc to take first {self.k}, but got {len(resps[0])} only! Please increase TaskConfig.repeats ."
+        return map(lambda r: r[: self.k], resps)
 
 
 class MajorityVoteFilter(Filter):
@@ -40,6 +42,7 @@ class MajorityVoteFilter(Filter):
         Each entry of `resps` is a list of model responses.
         We select the response that occurs most frequently in each entry of `resps`.
         """
+
         def select_majority(resp):
             counts = Counter(resp)
             vote = counts.most_common(1)[0][0]