code: def modify_block_args(block_args, kernel_size, exp_ratio):
      block_type = block_args["block_type"]
      if block_type == "cn":
          block_args["kernel_size"] = kernel_size
      elif block_type == "er":
          block_args["exp_kernel_size"] = kernel_size
      else:
          block_args["dw_kernel_size"] = kernel_size
  
      if block_type == "ir" or block_type == "er":
          block_args["exp_ratio"] = exp_ratio
      return block_args
  
file_path: nni/examples/nas/legacy/cream/lib/utils/builder_util.py
imports: import re
  import math
  import torch.nn
  from copy import deepcopy
  from timm.utils import *
  from timm.models.layers.activations import Swish
  from timm.models.layers import CondConv2d, get_condconv_initializer
label_desc: Use image node_id 3 for calling a global function with example usage: modify_block_args(block_args, kernel_size, exp_ratio) and returns: block_args