fix functional test

Author: roywei

src/sagemaker_training/process.py

@@ -256,23 +256,23 @@ def check_error(cmd, error_classes, processes_per_host, cwd=None, capture_error=
         if return_code == 137:
             extra_info = "OutOfMemory: Process killed by SIGKILL (signal 9)"
 
-        # default to user script error
-        error_class = errors.ExecuteUserScriptError
-        # use first found target error class if available
-        if stderr:
+        # throw internal error classes first
+        internal_errors = [err for err in dir(errors) if isclass(getattr(errors, err))]
+        error_class = next(
+            (name for name in error_classes if name in internal_errors), "ExecuteUserScriptError"
+        )
+        error_class = getattr(errors, error_class)
+
+        # only replace ExecuteUserScriptError with custom library errors
+        if stderr and error_class == errors.ExecuteUserScriptError:
             # find the first target error in stderr
             error_name = next((str(name) for name in error_classes if str(name) in stderr), False)
             if error_name:
-                # if error name is one of toolkit errors
-                if str(error_name) in [x for x in dir(errors) if isclass(getattr(errors, x))]:
-                    error_class = getattr(errors, error_name)
-                else:
-                    # if error is one of custom errors
-                    error_class = type(
-                        error_name,
-                        (errors._CalledProcessError,),  # pylint: disable=protected-access
-                        {},
-                    )
+                error_class = type(
+                    error_name,
+                    (errors._CalledProcessError,),  # pylint: disable=protected-access
+                    {},
+                )
 
         raise error_class(
             cmd=" ".join(cmd) if isinstance(cmd, list) else cmd,