code: def _foo(model):
      assert isinstance(model, MNISTModel)
  
imports: import pytest
  import pytorch_lightning
  import torch
  import torch.nn
  import torch.nn.functional
  from sklearn.datasets import load_diabetes
  from torch.utils.data import Dataset
  from torchvision import transforms
  from torchvision.datasets import MNIST
  import nni
  import nni.nas.evaluator.pytorch.lightning
  from nni.nas.evaluator import FunctionalEvaluator, Evaluator
  from nni.nas.space import RawFormatModelSpace
label_desc: Use image node_id 2 for calling a global function with example usage: _foo(model) without return types