Skip to content

Commit 5105475

Browse files
authored
Fix logging bug in oneshot.py (#1213)
SUMMARY: Changed `logger.waning` to `logger.warning` in `oneshot.py` TEST PLAN: `make test` passes Signed-off-by: Aman Gupta <aman2304@gmail.com>
1 parent 203c9b7 commit 5105475

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/llmcompressor/entrypoints/oneshot.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -361,7 +361,7 @@ def _get_output_dir_from_argv() -> Optional[str]:
361361

362362
# raise depreciation warnings
363363
if data_args.remove_columns is not None:
364-
logger.waning(
364+
logger.warning(
365365
"`remove_columns` argument is depreciated. When tokenizing datasets, all "
366366
"columns which are invalid inputs the tokenizer will be removed",
367367
DeprecationWarning,

0 commit comments

Comments
 (0)