Skip to content

Commit dc5a243

Browse files
author
George
authored
Merge branch 'main' into datasetargs
2 parents 5d31244 + 5105475 commit dc5a243

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/llmcompressor/entrypoints/oneshot.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -365,8 +365,8 @@ def _get_output_dir_from_argv() -> Optional[str]:
365365
recipe_args.recipe_args = arg_dict
366366

367367
# raise depreciation warnings
368-
if dataset_args.remove_columns is not None:
369-
logger.waning(
368+
if data_args.remove_columns is not None:
369+
logger.warning(
370370
"`remove_columns` argument is depreciated. When tokenizing datasets, all "
371371
"columns which are invalid inputs the tokenizer will be removed",
372372
DeprecationWarning,

0 commit comments

Comments
 (0)