code: def align_data_set(dataset):
      out_types = get_output_types(
          dataset.data,
          dataset.label,
          dataset.sample_weight,
          dataset.init_score,
      )
      op = LGBMAlign(
          data=dataset.data,
          label=dataset.label,
          sample_weight=dataset.sample_weight,
          init_score=dataset.init_score,
          output_types=out_types,
      )
      return op()
  
file_path: mars/mars/learn/contrib/lightgbm/_align.py
imports: from ....None import opcodes
  from ....core import ExecutableTuple, get_output_types, recursive_tile
  from ....serialization.serializables import AnyField
  from ....utils import has_unknown_shape
  from ...operands import LearnOperand, LearnOperandMixin
label_desc: Use image node_id 1 for calling a global function with example usage: align_data_set(dataset) and returns: op