53f1068a5dfa4fc271107955c304b81b7ddf1380,coremltools/converters/tensorflow/test/test_tf_2x.py,TestModelFormats,test_subclassed_keras_model,#TestModelFormats#,170

Before Change

        model = coremltools.converters.tensorflow.convert(
            self.saved_model_dir,
            inputs={input_name: (4, 4)},
            outputs=["Identity"],
            target_ios="13"
        )
        assert isinstance(model, coremltools.models.MLModel)

After Change

        keras_model._set_inputs(inputs)
        keras_model.save(self.saved_model_dir, save_format="tf")
        input_name = keras_model.inputs[0].name.split(":")[0]
        output_name = keras_model.outputs[0].name.split(":")[0].split("/")[-1]
        // convert and validate
        model = coremltools.converters.tensorflow.convert(
            self.saved_model_dir,
            inputs={input_name: (4, 4)},
            outputs=[output_name])
        assert isinstance(model, coremltools.models.MLModel)
        self._test_prediction(keras_model=keras_model, core_ml_model=model, inputs=inputs)

Instances