code: def graph_classification():
      """Classify graphs."""
      cli = FlashCLI(
          GraphClassifier,
          GraphClassificationData,
          default_datamodule_builder=from_tu_dataset,
          default_arguments={
              "trainer.max_epochs": 3,
          },
          finetune=False,
          datamodule_attributes={
              "num_classes",
              "labels",
              "num_features",
          },
      )
  
      cli.trainer.save_checkpoint("graph_classification.pt")
  
imports: from flash.core.utilities.flash_cli import FlashCLI
  from flash.graph import GraphClassificationData, GraphClassifier
label_desc: Use image node_id 2 for calling a global function with example usage: graph_classification() without return types