set disable_profiler= True

Author: knikure

src/sagemaker/estimator.py

@@ -595,7 +595,7 @@ def __init__(
 
         if not _region_supports_profiler(
             self.sagemaker_session.boto_region_name
-        ) or not _instance_type_supports_profiler(self.instance_type):
+        ) or _instance_type_supports_profiler(self.instance_type):
             self.disable_profiler = True
 
         self.profiler_rule_configs = None

src/sagemaker/fw_utils.py

@@ -1077,8 +1077,8 @@ def _instance_type_supports_profiler(instance_type):
     if isinstance(instance_type, str):
         match = re.match(r"^ml[\._]([a-z\d]+)\.?\w*$", instance_type)
         if match and match[1].startswith("trn"):
-            return False
-    return True
+            return True
+    return False
 
 
 def validate_version_or_image_args(framework_version, py_version, image_uri):

tests/unit/test_fw_utils.py

@@ -1043,6 +1043,6 @@ def test_validate_unsupported_distributions_trainium_raises():
 
 
 def test_instance_type_supports_profiler():
-    assert fw_utils._instance_type_supports_profiler("ml.trn1.xlarge") is False
-    assert fw_utils._instance_type_supports_profiler("ml.m4.xlarge") is True
-    assert fw_utils._instance_type_supports_profiler("local") is True
+    assert fw_utils._instance_type_supports_profiler("ml.trn1.xlarge") is True
+    assert fw_utils._instance_type_supports_profiler("ml.m4.xlarge") is False
+    assert fw_utils._instance_type_supports_profiler("local") is False