code: def test_not_trainable(tmpdir):
      """Tests that the model gives an error when training, validating, or testing."""
      tudataset = datasets.TUDataset(root=tmpdir, name="KKI")
      model = GraphEmbedder(
          GraphClassifier(num_features=1, num_classes=1).backbone
      )
      datamodule = DataModule(
          GraphClassificationDatasetInput(
              RunningStage.TRAINING, tudataset
          ),
          GraphClassificationDatasetInput(
              RunningStage.VALIDATING, tudataset
          ),
          GraphClassificationDatasetInput(
              RunningStage.TESTING, tudataset
          ),
          transform=GraphClassificationInputTransform,
          batch_size=4,
      )
      trainer = Trainer(default_root_dir=tmpdir, num_sanity_val_steps=0)
      with pytest.raises(
          NotImplementedError,
          match="Training a `GraphEmbedder` is not supported.",
      ):
          trainer.fit(model, datamodule=datamodule)
  
      with pytest.raises(
          NotImplementedError,
          match="Validating a `GraphEmbedder` is not supported.",
      ):
          trainer.validate(model, datamodule=datamodule)
  
      with pytest.raises(
          NotImplementedError,
          match="Testing a `GraphEmbedder` is not supported.",
      ):
          trainer.test(model, datamodule=datamodule)
  
file_path: lightning-flash/tests/graph/embedding/test_model.py
imports: from typing import Any
  import pytest
  import torch
  from flash import RunningStage, Trainer
  from flash.core.data.data_module import DataModule
  from flash.core.utilities.imports import _TOPIC_GRAPH_AVAILABLE
  from flash.graph.classification.input import GraphClassificationDatasetInput
  from flash.graph.classification.input_transform import GraphClassificationInputTransform
  from flash.graph.classification.model import GraphClassifier
  from flash.graph.embedding.model import GraphEmbedder
  from torch import Tensor
  from tests.helpers.task_tester import TaskTester
label_desc: Use image node_id 2 for calling a global function with example usage: test_not_trainable(tmpdir) without return types