code: def get_block_fun(block_type):
      """Retrieves the block function by name."""
      block_funs = {
          "vanilla_block": VanillaBlock,
          "res_basic_block": ResBasicBlock,
          "res_bottleneck_block": ResBottleneckBlock,
          "res_bottleneck_linear_block": ResBottleneckLinearBlock,
      }
      err_str = "Block type '{}' not supported"
      assert block_type in block_funs.keys(), err_str.format(block_type)
      return block_funs[block_type]
  
imports: from pycls.core.config import cfg
  from pycls.models.blocks import activation, conv2d, conv2d_cx, gap2d, gap2d_cx, init_weights, linear, linear_cx, norm2d, norm2d_cx, pool2d, pool2d_cx, SE
  from torch.nn import Module
label_desc: Use image node_id 2 for calling a global function with example usage: get_block_fun(block_type) and returns: block_funs