code: def get_router(poly_config: PolyConfig) -> nn.Module:
      if poly_config.poly_type == "poly":
          return PolyRouter(poly_config)
      else:
          raise ValueError(
              f"Unsupported poly_type: {poly_config.poly_type}. "
              "Currently, only the following types are supported: "
              "`poly`."
          )
  
file_path: peft/src/peft/tuners/poly/router.py
imports: from abc import ABC, abstractmethod
  import torch
  from torch import nn
  from torch.distributions.relaxed_bernoulli import RelaxedBernoulli
  from .config import PolyConfig
label_desc: Use image node_id 1 for calling a global function with example usage: get_router(poly_config) and returns: PolyRouter