From 8bb38027c4ed381ea70f07e14bfa2483a5f8c94a Mon Sep 17 00:00:00 2001
From: Nora Belrose <belrose.nora@gmail.com>
Date: Tue, 18 Apr 2023 04:54:08 +0000
Subject: [PATCH] Remove vestigial label_column rename

---
 elk/extraction/prompt_loading.py | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/elk/extraction/prompt_loading.py b/elk/extraction/prompt_loading.py
index 744da86f..97d41c80 100644
--- 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,