Fix if condition

Author: Nikhil Kulkarni

src/sagemaker/model.py

@@ -279,11 +279,10 @@ def _compilation_image_uri(self, region, target_instance_type, framework, framew
 
         if framework == "xgboost":
             framework_suffix = "-neo"
+        elif target_instance_type.startswith("ml_inf"):
+            framework_prefix = "inferentia-"
         else:
-            if target_instance_type.startswith("ml_inf"):
-                framework_prefix = "inferentia-"
-            else:
-                framework_prefix = "neo-"
+            framework_prefix = "neo-"
 
         return image_uris.retrieve(
             "{}{}{}".format(framework_prefix, framework, framework_suffix),