code: def test_tensorflow_wrapper_keras_subclass_decorator_compile_args():
      import tensorflow as tf
  
      class UndecoratedModel(tf.keras.Model):
          def call(self, inputs):
              return inputs
  
      # Can't wrap an undecorated keras subclass model
      with pytest.raises(ValueError):
          TensorFlowWrapper(UndecoratedModel())
  
      @keras_subclass(
          "TestModel",
          X=numpy.array([0.0, 0.0]),
          Y=numpy.array([0.5]),
          input_shape=(2,),
          compile_args={"loss": "binary_crossentropy"},
      )
      class TestModel(tf.keras.Model):
          def call(self, inputs):
              return inputs
  
      model = TensorFlowWrapper(TestModel())
      model = model.from_bytes(model.to_bytes())
  
      assert model.shims[0]._model.loss == "binary_crossentropy"
      assert isinstance(model, Model)
  
imports: import numpy
  import pytest
  from thinc.api import Adam, ArgsKwargs, Linear, Model, TensorFlowWrapper, get_current_ops, keras_subclass, tensorflow2xp, xp2tensorflow
  from thinc.compat import has_cupy_gpu, has_tensorflow
  from thinc.util import to_categorical
  from ..util import check_input_converters, make_tempdir
label_desc: Use image node_id 16 for calling a global function with example usage: test_tensorflow_wrapper_keras_subclass_decorator_compile_args() and returns: inputs, inputs