code: def get_some_model(x_np, y_np, sanity_check=True):
      x_input = tf.keras.layers.Input(shape=x_np.shape[1:])
      y_outputs = tf.keras.layers.Lambda(lambda x: x)(x_input)
      base_model = tf.keras.Model(x_input, y_outputs)
  
      model = CRFModelWrapper(base_model, y_np.shape[-1])
  
      model.compile("adam")
      if sanity_check:
          model.fit(x=x_np, y=y_np)
          model.evaluate(x_np, y_np)
      model.predict(x_np)
      return model
  
file_path: addons/tensorflow_addons/text/tests/crf_wrapper_test.py
imports: import os
  import tempfile
  import numpy
  import pytest
  import tensorflow
  from tensorflow_addons.text.crf_wrapper import CRFModelWrapper
label_desc: Use image node_id 3 for calling a global function with example usage: get_some_model(x_np, y_np, sanity_check) and returns: model