Skip to content

Commit 78e9dea

Browse files
committed
fix bigbench config names
1 parent 8a95aa1 commit 78e9dea

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

datasets/bigbench/bigbench.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -91,6 +91,12 @@ def __init__(
9191
max_examples: Optional[int] = None,
9292
**kwargs,
9393
):
94+
if subtask_name is not None:
95+
name += f"_subtask={subtask_name}"
96+
if num_shots != 0:
97+
name += f"_num_shots={num_shots}"
98+
if max_examples is not None:
99+
name += f"_max_examples={max_examples}"
94100
super().__init__(
95101
name=name,
96102
**kwargs,

0 commit comments

Comments
 (0)