Skip to content

Commit e8798f4

Browse files
committed
fix: change id_range type
1 parent a810f31 commit e8798f4

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

bigcodebench/generate.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,7 @@ def run_codegen(
135135
strip_newlines: bool = False,
136136
direct_completion: bool = False,
137137
resume: bool = True,
138-
id_range: Tuple[int, int] = None,
138+
id_range: str = None,
139139
backend: str = "vllm",
140140
base_url: str = None,
141141
tp: int = 1,
@@ -152,6 +152,7 @@ def run_codegen(
152152
print("Greedy decoding ON (--greedy): setting n_samples=1, temperature=0")
153153

154154
if id_range is not None:
155+
id_range = [int(i) for i in id_range.split("-")]
155156
assert len(id_range) == 2, "id_range must be a list of length 2"
156157
assert id_range[0] < id_range[1], "id_range must be increasing"
157158
id_range = tuple(id_range)

0 commit comments

Comments
 (0)