Remove vestigial label_column rename

michaelbyun · Apr 18, 2023 · 8bb3802 · 8bb3802
1 parent e1b1834
commit 8bb3802
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/elk/extraction/prompt_loading.py b/elk/extraction/prompt_loading.py
@@ -178,9 +178,6 @@ def load_prompts(
  extra_cols = list(assert_type(Features, ds.features))
  extra_cols.remove(label_column)
 
- if label_column != "label":
- ds = ds.rename_column(label_column, "label")
-
  for example in ds:
  yield _convert_to_prompts(
  example,