syntax

Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
vllm-project · Dec 20, 2024 · bbe3eb6 · bbe3eb6
1 parent 5b4ee22
commit bbe3eb6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/llmcompressor/transformers/finetune/data/base.py b/src/llmcompressor/transformers/finetune/data/base.py
@@ -240,7 +240,7 @@ def filter_processor_args(self, dataset: DatasetType) -> DatasetType:
 
         column_names = get_columns(dataset)
         return dataset.remove_columns(
-            list(set(column_names) - set(processor_kwargs)) - set([self.PROMPT_KEY])
+            list(set(column_names) - set(processor_kwargs) - set([self.PROMPT_KEY]))
         )
 
     def tokenize(self, data: LazyRow) -> Dict[str, Any]: