code: def test_nasnet_corner_case():
      # The case is that output channel of reduce cell and normal cell are different
      # CellPreprocessor needs to know whether its predecessors are normal cell / reduction cell
      arch = {
          "width": 32,
          "depth": 8,
          "normal/op_2_0": "max_pool_7x7",
          "normal/op_2_1": "conv_1x1",
          "normal/op_3_0": "sep_conv_5x5",
          "normal/op_3_1": "max_pool_7x7",
          "normal/op_4_0": "sep_conv_5x5",
          "normal/op_4_1": "conv_1x1",
          "normal/op_5_0": "max_pool_3x3",
          "normal/op_5_1": "sep_conv_5x5",
          "normal/op_6_0": "max_pool_7x7",
          "normal/op_6_1": "sep_conv_5x5",
          "normal/input_2_0": [0],
          "normal/input_2_1": [0],
          "normal/input_3_0": [0],
          "normal/input_3_1": [1],
          "normal/input_4_0": [1],
          "normal/input_4_1": [2],
          "normal/input_5_0": [0],
          "normal/input_5_1": [1],
          "normal/input_6_0": [0],
          "normal/input_6_1": [2],
          "reduce/op_2_0": "dil_conv_3x3",
          "reduce/op_2_1": "max_pool_7x7",
          "reduce/op_3_0": "dil_conv_3x3",
          "reduce/op_3_1": "dil_conv_3x3",
          "reduce/op_4_0": "conv_7x1_1x7",
          "reduce/op_4_1": "conv_7x1_1x7",
          "reduce/op_5_0": "max_pool_3x3",
          "reduce/op_5_1": "conv_1x1",
          "reduce/op_6_0": "sep_conv_7x7",
          "reduce/op_6_1": "sep_conv_3x3",
          "reduce/input_2_0": [1],
          "reduce/input_2_1": [1],
          "reduce/input_3_0": [0],
          "reduce/input_3_1": [1],
          "reduce/input_4_0": [2],
          "reduce/input_4_1": [1],
          "reduce/input_5_0": [0],
          "reduce/input_5_1": [4],
          "reduce/input_6_0": [3],
          "reduce/input_6_1": [3],
      }
  
      _test_searchspace_on_dataset(searchspace.NASNet(), arch=arch)
  
file_path: nni/test/algo/nas/hub/test_sanity.py
imports: from torchvision import transforms
  from torchvision.datasets import FakeData
  import pytest
  import pytorch_lightning
  import nni
  import nni.nas.evaluator.pytorch.lightning
  import nni.nas.hub.pytorch
  from nni.mutable import ConstraintViolation
  from nni.nas.nn.pytorch import ModelSpace
label_desc: Use image node_id 5 for calling a global function with example usage: test_nasnet_corner_case() without return types