Merge branch 'main' into spar_mt_prompt_invar

michaelbyun · Apr 12, 2023 · 715bba8 · 715bba8
2 parents d91acac + 89b2346
commit 715bba8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/elk/extraction/prompt_loading.py b/elk/extraction/prompt_loading.py
@@ -225,7 +225,7 @@ def _convert_to_prompts(
  fewshot_iter: Optional[Iterator[list[dict]]] = None,
 ) -> dict[str, Any]:
  """Prompt-generating function to pass to `IterableDataset.map`."""
- label = assert_type(int, example[label_column])
+ assert_type(int, example[label_column])
  prompts = []
  templates = list(prompter.templates.values())
  if num_variants < len(templates):