File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
fast_llm/data/preparator/gpt_memmap Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -112,7 +112,7 @@ def _validate(self) -> None:
112
112
113
113
@config_class
114
114
class FieldCombinePreparatorConfig (Config ):
115
- fields : typing .List [str ] = Field (
115
+ col_names : typing .List [str ] = Field (
116
116
default_factory = list ,
117
117
desc = "Fields of the dataset to combine." ,
118
118
hint = FieldHint .core ,
Original file line number Diff line number Diff line change @@ -211,11 +211,11 @@ def run(self) -> None:
211
211
212
212
# Check for combining fields
213
213
if self ._config .combine_fields :
214
- Assert .eq (len (set (self ._config .combine_fields .fields ).intersection (dataset .column_names )), len (self ._config .combine_fields .fields ))
214
+ Assert .eq (len (set (self ._config .combine_fields .col_names ).intersection (dataset .column_names )), len (self ._config .combine_fields .col_names ))
215
215
dataset = dataset .map (
216
216
lambda example : {
217
217
self ._config .combine_fields .new_field_name : self ._config .combine_fields .delimiter .join (
218
- str (example [column ]) for column in self ._config .combine_fields .fields
218
+ str (example [column ]) for column in self ._config .combine_fields .col_names
219
219
)
220
220
},
221
221
batched = False ,
You can’t perform that action at this time.
0 commit comments