fix: disable Debugger defaults in unsupported regions (aws#1272)

Author: knakad

src/sagemaker/estimator.py

@@ -38,6 +38,7 @@
     parse_s3_url,
     UploadedCode,
     validate_source_dir,
+    _region_supports_debugger,
 )
 from sagemaker.job import _Job
 from sagemaker.local import LocalSession
@@ -1674,7 +1675,9 @@ def _validate_and_set_debugger_configs(self):
         """
         Set defaults for debugging
         """
-        if self.debugger_hook_config is None:
+        if self.debugger_hook_config is None and _region_supports_debugger(
+            self.sagemaker_session.boto_region_name
+        ):
             self.debugger_hook_config = DebuggerHookConfig(s3_output_path=self.output_path)
         elif not self.debugger_hook_config:
             self.debugger_hook_config = None

src/sagemaker/fw_utils.py

@@ -84,6 +84,8 @@
     "pytorch-serving": [1, 2, 0],
 }
 
+DEBUGGER_UNSUPPORTED_REGIONS = ["us-gov-west-1", "us-iso-east-1"]
+
 
 def is_version_equal_or_higher(lowest_version, framework_version):
     """Determine whether the ``framework_version`` is equal to or higher than
@@ -504,3 +506,16 @@ def python_deprecation_warning(framework, latest_supported_version):
     return PYTHON_2_DEPRECATION_WARNING.format(
         framework=framework, latest_supported_version=latest_supported_version
     )
+
+
+def _region_supports_debugger(region_name):
+    """Returns boolean indicating whether the region supports Amazon SageMaker Debugger.
+
+    Args:
+        region_name (str): Name of the region to check against.
+
+    Returns:
+        bool: Whether or not the region supports Amazon SageMaker Debugger.
+
+    """
+    return region_name.lower() not in DEBUGGER_UNSUPPORTED_REGIONS

src/sagemaker/tensorflow/estimator.py

@@ -723,7 +723,9 @@ def _validate_and_set_debugger_configs(self):
                 )
             self.debugger_hook_config = None
             self.debugger_rule_configs = None
-        elif self.debugger_hook_config is None:
+        elif self.debugger_hook_config is None and fw._region_supports_debugger(
+            self.sagemaker_session.boto_session.region_name
+        ):
             # Set defaults for debugging.
             self.debugger_hook_config = DebuggerHookConfig(s3_output_path=self.output_path)
 

tests/unit/test_fw_utils.py

@@ -1035,3 +1035,13 @@ def test_model_code_key_prefix_with_all_none_fail():
     with pytest.raises(TypeError) as error:
         fw_utils.model_code_key_prefix(None, None, None)
     assert "expected string" in str(error)
+
+
+def test_region_supports_debugger_feature_returns_true_for_supported_regions():
+    assert fw_utils._region_supports_debugger("us-west-2") is True
+    assert fw_utils._region_supports_debugger("us-east-2") is True
+
+
+def test_region_supports_debugger_feature_returns_false_for_unsupported_regions():
+    assert fw_utils._region_supports_debugger("us-gov-west-1") is False
+    assert fw_utils._region_supports_debugger("us-iso-east-1") is False