code: def cli_main():
      cli = LightningCLI(
          LitClassifier,
          MyDataModule,
          seed_everything_default=1234,
          save_config_kwargs={"overwrite": True},
          run=False,
      )
      cli.trainer.fit(cli.model, datamodule=cli.datamodule)
      cli.trainer.test(ckpt_path="best", datamodule=cli.datamodule)
      predictions = cli.trainer.predict(
          ckpt_path="best", datamodule=cli.datamodule
      )
      print(predictions[0])
  
imports: from os import path
  from typing import Optional
  import torch
  from lightning.pytorch import LightningDataModule, LightningModule, cli_lightning_logo
  from lightning.pytorch.cli import LightningCLI
  from lightning.pytorch.demos.mnist_datamodule import MNIST
  from lightning.pytorch.utilities.imports import _TORCHVISION_AVAILABLE
  from torch.nn import functional
  from torch.utils.data import DataLoader, random_split
label_desc: Use image node_id 1 for calling a global function with example usage: cli_main() without return types