Skip to content

Commit f30dda1

Browse files
authored
use strip() instead of lstrip() in remove_whitespace filter (#3408)
1 parent 2bb3aee commit f30dda1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lm_eval/filters/extraction.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -105,13 +105,13 @@ def filter_set(inst):
105105

106106
@register_filter("remove_whitespace")
107107
class WhitespaceFilter(Filter):
108-
"""Filters out leading whitespace from responses."""
108+
"""Filters out leading and trailing whitespace from responses."""
109109

110110
def apply(self, resps: list[list[str]], docs: list[dict]) -> list[list[str]]:
111111
def filter_set(inst):
112112
filtered_resp = []
113113
for resp in inst:
114-
resp = resp.lstrip()
114+
resp = resp.strip()
115115
filtered_resp.append(resp)
116116
return filtered_resp
117117

0 commit comments

Comments
 (0)