Skip to content

Commit 459c913

Browse files
committed
Fix dataset when swt-filter = False
1 parent c4a0490 commit 459c913

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/dataset.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -171,7 +171,7 @@ def load_swebench_dataset(name="princeton-nlp/SWE-bench", split="test", is_swt:
171171
if filter_swt:
172172
return [i for i in instances if i["instance_id"] not in instance_filter]
173173
else:
174-
instances
174+
return instances
175175

176176

177177
### MARK - Patch Correction
@@ -183,4 +183,4 @@ def load_swebench_dataset(name="princeton-nlp/SWE-bench", split="test", is_swt:
183183
PATCH_HUNK_PATTERN = re.compile(
184184
r"\@\@\s+\-(\d+),(\d+)\s+\+(\d+),(\d+)\s+\@\@(.+?)(?=diff\ |\-\-\-\ a\/|\@\@\ \-|\Z)",
185185
re.DOTALL,
186-
)
186+
)

0 commit comments

Comments
 (0)