code: def test_repeat():
      net = nas_nn.Repeat(
          nn.Linear(3, 3, bias=False), nni.choice("rep", [0, 1, 2, 4])
      )
      profiler = FlopsParamsProfiler(net, torch.randn(1, 3))
      assert profiler.profile({"rep": 0}).flops == 0
      assert profiler.profile({"rep": 1}).flops == 9
      assert profiler.profile({"rep": 2}).flops == 18
      assert profiler.profile({"rep": 4}).flops == 36
      assert profiler.profile({"rep": 2}).params == 18
  
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 7 for calling a global function with example usage: test_repeat() without return types