Spaces:
Runtime error
Runtime error
shlomihod
commited on
Commit
·
2a72d30
1
Parent(s):
fc5b076
fix lower column name bug
Browse files
app.py
CHANGED
@@ -123,7 +123,8 @@ def prepare_datasets(dataset_name):
|
|
123 |
input_columns = []
|
124 |
for input_column in original_input_columns:
|
125 |
lowered_input_column = input_column.lower()
|
126 |
-
|
|
|
127 |
input_columns.append(lowered_input_column)
|
128 |
|
129 |
ds = ds["train"].train_test_split(
|
|
|
123 |
input_columns = []
|
124 |
for input_column in original_input_columns:
|
125 |
lowered_input_column = input_column.lower()
|
126 |
+
if input_column != lowered_input_column:
|
127 |
+
ds = ds.rename_column(input_column, lowered_input_column)
|
128 |
input_columns.append(lowered_input_column)
|
129 |
|
130 |
ds = ds["train"].train_test_split(
|