code: def test_inputchoice():
      net = nas_nn.InputChoice(3, 1, label="a")
      profiler = FlopsParamsProfiler(
          net, [torch.randn(3), torch.randn(3)]
      )
      assert profiler.profile({"a": [1]}).flops == 0
  
file_path: nni/test/ut/nas/profiler/test_flops.py
imports: import pytest
  import torch
  from torch import nn
  import nni
  import nni.nas.nn.pytorch
  from nni.mutable import Categorical
  from nni.nas.profiler.pytorch.flops import FlopsParamsProfiler, FlopsProfiler, NumParamsProfiler, FlopsResult, register_flops_formula
  from ut.nas.nn.models import MODELS
label_desc: Use image node_id 6 for calling a global function with example usage: test_inputchoice() without return types