Update prompt_loading.py

Remove print label
michaelbyun · Mar 28, 2023 · 9186326 · 9186326
1 parent d2c66b0
commit 9186326
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/elk/extraction/prompt_loading.py b/elk/extraction/prompt_loading.py
@@ -204,7 +204,6 @@ def _convert_to_prompts(
  fewshot_iter: Optional[Iterator[list[dict]]] = None,
 ) -> dict[str, Any]:
  """Prompt-generating function to pass to `IterableDataset.map`."""
- print(f"label: {example[label_column]}")
  prompts = []
  templates = list(prompter.templates.values())
  if num_variants < len(templates):