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