def _validate_and_set_debugger_configs(self):
"""Disable Debugger Hook Config for ParameterServer (PS) as it is not supported in smdebug.
Else, set default HookConfig
"""
ps_enabled = "parameter_server" in self.distribution and self.distribution[
"parameter_server"].get("enabled", False)
if ps_enabled:
if self.debugger_hook_config is not None or self.debugger_rule_configs is not None:
logger.info(
"Amazon SageMaker Debugger does not currently support "
"Parameter Server distribution")
self.debugger_hook_config = None
self.debugger_rule_configs = 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)
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
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
