Skip to content

Commit 52b9179

Browse files
committed
rename fields to colnames name re-used
1 parent 9c371ab commit 52b9179

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

fast_llm/data/preparator/gpt_memmap/config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ def _validate(self) -> None:
112112

113113
@config_class
114114
class FieldCombinePreparatorConfig(Config):
115-
fields: typing.List[str] = Field(
115+
col_names: typing.List[str] = Field(
116116
default_factory=list,
117117
desc="Fields of the dataset to combine.",
118118
hint=FieldHint.core,

fast_llm/data/preparator/gpt_memmap/prepare.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -211,11 +211,11 @@ def run(self) -> None:
211211

212212
# Check for combining fields
213213
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))
215215
dataset = dataset.map(
216216
lambda example: {
217217
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
219219
)
220220
},
221221
batched=False,

0 commit comments

Comments
 (0)