fix

Author: laurenyu

src/sagemaker/cli/compatibility/v2/ast_transformer.py

@@ -19,6 +19,7 @@
 
 FUNCTION_CALL_MODIFIERS = [
     modifiers.renamed_params.EstimatorImageURIRenamer(),
+    modifiers.renamed_params.ModelImageURIRenamer(),
     modifiers.framework_version.FrameworkVersionEnforcer(),
     modifiers.tf_legacy_mode.TensorFlowLegacyModeConstructorUpgrader(),
     modifiers.tf_legacy_mode.TensorBoardParameterRemover(),