SageMaker Neo Compilation Input Error - Converting Tensflor to CoreML

Hi all,

I have a model that I've finetuned from the TF MobileResnet model available in SageMaker. The model is working well, and returns correct inference values when I host it using the SageMaker inference endpoints.

I am now trying to convert the model to coreml, however when attempting to run this job via SageMaker Neo I am receiving the following error:

ClientError: InputConfiguration: Unable to determine the type of the model, i.e. the source framework. Please provide the value of argument "source", from one of ["tensorflow", "pytorch", "mil"]. Note that model conversion requires the source package that generates the model. Please make sure you have the appropriate version of source package installed. E.g., if you\'re converting model originally trained with TensorFlow 1.14, make sure you have `tensorflow==1.14` installed. tensorflow 2.2.0, coremltools 4.0 For further troubleshooting common failures please visit: https://docs.aws.amazon.com/sagemaker/latest/dg/neo-troubleshooting-compilation.html

I cannot find any information about this error in the docs linked in the error message. Any help would be greatly appreciated.