code: def get_model(W_b_input, cls=Linear):
      W, b, input_ = W_b_input
      nr_out, nr_in = W.shape
      model = cls(nr_out, nr_in)
      model.set_param("W", W)
      model.set_param("b", b)
      model.initialize()
      return model
  
file_path: thinc/thinc/tests/util.py
imports: import contextlib
  import shutil
  import tempfile
  from pathlib import Path
  import numpy
  import pytest
  from thinc.api import ArgsKwargs, Linear, Padded, Ragged
  from thinc.util import has_cupy, is_cupy_array, is_numpy_array
label_desc: Use image node_id 2 for calling a global function with example usage: get_model(W_b_input, cls) and returns: model