fix multiclass label bug

michaelbyun · Apr 12, 2023 · 66c7a6b · 66c7a6b
1 parent e536aa6
commit 66c7a6b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/elk/extraction/prompt_loading.py b/elk/extraction/prompt_loading.py
@@ -259,7 +259,7 @@ def qa_cat(q: str, a: str) -> str:
  raise ValueError(f'Prompt duplicated {dup_count} times! "{maybe_dup}"')
 
  return dict(
- label=label,
+ label=new_label,
  prompts=prompts,
  template_names=prompter.all_template_names,
  )