class_docstr: Wrap the data collator to remove unused columns before they are passed to the collator.
class_docstr_tok: ["Wrap","the","data","collator","to","remove","unused","columns","before","they","are","passed","to","the","collator","."]
code: def __init__(
      self,
      data_collator,
      signature_columns,
      logger=None,
      model_name: Optional[str] = None,
      description: Optional[str] = None,
  ):
      self.data_collator = data_collator
      self.signature_columns = signature_columns
      self.logger = logger
      self.description = description
      self.model_name = model_name
      self.message_logged = False
  
imports: import copy
  import functools
  import gc
  import inspect
  import os
  import random
  import re
  import threading
  import time
  from typing import Any, Dict, List, NamedTuple, Optional, Tuple, Union
  import numpy
  from .utils import ExplicitEnum, is_psutil_available, is_tf_available, is_torch_available, is_torch_cuda_available, is_torch_mps_available, is_torch_npu_available, is_torch_tpu_available, is_torch_xpu_available, requires_backends
label_desc: Use image node_id 1 to create a new RemoveColumnsCollator object  with example: obj = RemoveColumnsCollator(data_collator, signature_columns, logger, model_name, description)