fix merge main

Author: George Ohashi

src/llmcompressor/entrypoints/oneshot.py

@@ -365,7 +365,7 @@ def _get_output_dir_from_argv() -> Optional[str]:
             recipe_args.recipe_args = arg_dict
 
     # raise depreciation warnings
-    if data_args.remove_columns is not None:
+    if dataset_args.remove_columns is not None:
         logger.warning(
             "`remove_columns` argument is depreciated. When tokenizing datasets, all "
             "columns which are invalid inputs the tokenizer will be removed",