code: def import_lightning():  # noqa: F402
      try:
          import lightning.pytorch as pl
      except ModuleNotFoundError:
          import pytorch_lightning as pl
      return pl
  
imports: import logging
  import os
  import shutil
  import tempfile
  from typing import Any, Dict
  import torch
  from packaging.version import Version
  import ray
  from ray import train
  from ray._private.usage.usage_lib import TagKey, record_extra_usage_tag
  from ray.train import Checkpoint
  from ray.util import PublicAPI
label_desc: Use image node_id 1 for calling a global function with example usage: import_lightning() and returns: pl