shlomihod commited on
Commit
2a72d30
·
1 Parent(s): fc5b076

fix lower column name bug

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- ds = ds.rename_column(input_column, lowered_input_column)
 
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(