code: def get_ray_tune_ckpt_callback():
      """
      This is a workaround for the issue caused by the mixed use of old and new lightning's import style.
      https://github.com/optuna/optuna/issues/4689
      We can remove this function after ray adopts the new lightning import style.
      """
      from ray.tune.integration.pytorch_lightning import (
          TuneReportCheckpointCallback,
      )
  
      class _TuneReportCheckpointCallback(
          TuneReportCheckpointCallback, pl.Callback
      ):
          def __init__(self, *args, **kwargs):
              super().__init__(*args, **kwargs)
  
      return _TuneReportCheckpointCallback
  
file_path: autogluon/multimodal/src/autogluon/multimodal/utils/hpo.py
imports: import logging
  import os
  import shutil
  import lightning.pytorch
  import yaml
  from autogluon.common.utils.context import set_torch_num_threads
  from ..constants import BEST_K_MODELS_FILE, RAY_TUNE_CHECKPOINT
  from .matcher import create_siamese_model
  from .model import create_fusion_model
label_desc: Use image node_id 1 for calling a global function with example usage: get_ray_tune_ckpt_callback() and returns: _TuneReportCheckpointCallback