handle frame in stack trace that doesn't have a module

hydra/_internal/utils.py

@@ -7,7 +7,7 @@
 from dataclasses import dataclass
 from os.path import dirname, join, normpath, realpath
 from traceback import print_exc, print_exception
-from types import FrameType
+from types import FrameType, TracebackType
 from typing import Any, Callable, List, Optional, Sequence, Tuple, Union
 
 from omegaconf.errors import OmegaConfBaseException
@@ -213,75 +213,79 @@ def run_and_report(func: Any) -> Any:
         if _is_env_set("HYDRA_FULL_ERROR") or is_under_debugger():
             raise ex
         else:
-            if isinstance(ex, CompactHydraException):
-                sys.stderr.write(str(ex) + os.linesep)
-                if isinstance(ex.__cause__, OmegaConfBaseException):
-                    sys.stderr.write(str(ex.__cause__) + os.linesep)
-            else:
-                # Custom printing that strips the Hydra related stack frames from the top
-                # And any omegaconf frames from the bottom.
-                # It is possible to add additional libraries to sanitize from the bottom later,
-                # maybe even make it configurable.
-
-                tb: Any = ex.__traceback__
-                search_max = 10
-                # strip Hydra frames from start of stack
-                # will strip until it hits run_job()
-                while search_max > 0:
-                    if tb is None:
-                        break
-                    frame = tb.tb_frame
-                    tb = tb.tb_next
-                    search_max = search_max - 1
-                    if inspect.getframeinfo(frame).function == "run_job":
-                        break
-
-                if search_max == 0 or tb is None:
-                    # could not detect run_job, probably a runtime exception before we got there.
-                    # do not sanitize the stack trace.
-                    print_exc()
-                    sys.exit(1)
-
-                # strip OmegaConf frames from bottom of stack
-                end = tb
-                num_frames = 0
-                while end is not None:
-                    frame = end.tb_frame
-                    mdl = inspect.getmodule(frame)
-                    assert mdl is not None
-                    name = mdl.__name__
-                    if name.startswith("omegaconf."):
-                        break
-                    end = end.tb_next
-                    num_frames = num_frames + 1
-
-                @dataclass
-                class FakeTracebackType:
-                    tb_next: Any = None  # Optional["FakeTracebackType"]
-                    tb_frame: Optional[FrameType] = None
-                    tb_lasti: Optional[int] = None
-                    tb_lineno: Optional[int] = None
-
-                iter_tb = tb
-                final_tb = FakeTracebackType()
-                cur = final_tb
-                added = 0
-                while True:
-                    cur.tb_lasti = iter_tb.tb_lasti
-                    cur.tb_lineno = iter_tb.tb_lineno
-                    cur.tb_frame = iter_tb.tb_frame
-
-                    if added == num_frames - 1:
-                        break
-                    added = added + 1
-                    cur.tb_next = FakeTracebackType()
-                    cur = cur.tb_next
-                    iter_tb = iter_tb.tb_next
-
-                print_exception(etype=None, value=ex, tb=final_tb)  # type: ignore
-            sys.stderr.write(
-                "\nSet the environment variable HYDRA_FULL_ERROR=1 for a complete stack trace.\n"
-            )
+            try:
+                if isinstance(ex, CompactHydraException):
+                    sys.stderr.write(str(ex) + os.linesep)
+                    if isinstance(ex.__cause__, OmegaConfBaseException):
+                        sys.stderr.write(str(ex.__cause__) + os.linesep)
+                else:
+                    # Custom printing that strips the Hydra related stack frames from the top
+                    # And any omegaconf frames from the bottom.
+                    # It is possible to add additional libraries to sanitize from the bottom later,
+                    # maybe even make it configurable.
+
+                    tb = ex.__traceback__
+                    search_max = 10
+                    # strip Hydra frames from start of stack
+                    # will strip until it hits run_job()
+                    while search_max > 0:
+                        if tb is None:
+                            break
+                        frame = tb.tb_frame
+                        tb = tb.tb_next
+                        search_max = search_max - 1
+                        if inspect.getframeinfo(frame).function == "run_job":
+                            break
+
+                    if search_max == 0 or tb is None:
+                        # could not detect run_job, probably a runtime exception before we got there.
+                        # do not sanitize the stack trace.
+                        print_exc()
+                        sys.exit(1)
+
+                    # strip OmegaConf frames from bottom of stack
+                    end: Optional[TracebackType] = tb
+                    num_frames = 0
+                    while end is not None:
+                        frame = end.tb_frame
+                        mdl = inspect.getmodule(frame)
+                        assert mdl is not None
+                        name = mdl.__name__
+                        if name.startswith("omegaconf."):
+                            break
+                        end = end.tb_next
+                        num_frames = num_frames + 1
+
+                    @dataclass
+                    class FakeTracebackType:
+                        tb_next: Any = None  # Optional["FakeTracebackType"]
+                        tb_frame: Optional[FrameType] = None
+                        tb_lasti: Optional[int] = None
+                        tb_lineno: Optional[int] = None
+
+                    iter_tb = tb
+                    final_tb = FakeTracebackType()
+                    cur = final_tb
+                    added = 0
+                    while True:
+                        cur.tb_lasti = iter_tb.tb_lasti
+                        cur.tb_lineno = iter_tb.tb_lineno
+                        cur.tb_frame = iter_tb.tb_frame
+
+                        if added == num_frames - 1:
+                            break
+                        added = added + 1
+                        cur.tb_next = FakeTracebackType()
+                        cur = cur.tb_next
+                        assert iter_tb.tb_next is not None
+                        iter_tb = iter_tb.tb_next
+
+                    print_exception(etype=None, value=ex, tb=final_tb)  # type: ignore
+                sys.stderr.write(
+                    "\nSet the environment variable HYDRA_FULL_ERROR=1 for a complete stack trace.\n"
+                )
+            except Exception:
+                print_exception(etype=None, value=ex, tb=ex.__traceback__)
         sys.exit(1)
 
 

tests/test_utils.py

@@ -1,14 +1,19 @@
 # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
+import io
 import os
 from pathlib import Path
+from textwrap import dedent
 from typing import Any
+from unittest.mock import patch
 
 from omegaconf import DictConfig, OmegaConf
-from pytest import mark, raises
+from pytest import mark, param, raises
 
 from hydra import utils
+from hydra._internal.utils import run_and_report
 from hydra.conf import HydraConf, RuntimeConf
 from hydra.core.hydra_config import HydraConfig
+from hydra.test_utils.test_utils import assert_regex_match
 
 
 def test_get_original_cwd(hydra_restore_singletons: Any) -> None:
@@ -60,3 +65,128 @@ def test_to_absolute_path_without_hydra(
     path = str(Path(path))
     expected = str(Path(expected).absolute())
     assert utils.to_absolute_path(path) == expected
+
+
+class TestRunAndReport:
+    class DemoFunctions:
+        """Demo inputs for `run_and_report`"""
+
+        @staticmethod
+        def success_func() -> Any:
+            return 123
+
+        @staticmethod
+        def simple_error() -> None:
+            assert False, "simple_err_msg"
+
+        @staticmethod
+        def run_job_wrapper() -> None:
+            def run_job() -> None:
+                def nested_error() -> None:
+                    assert False, "nested_err"
+
+                nested_error()
+
+            run_job()
+
+        @staticmethod
+        def omegaconf_job_wrapper() -> None:
+            def run_job() -> None:
+                def job_calling_omconf() -> None:
+                    from omegaconf import OmegaConf
+
+                    OmegaConf.resolve(123)  # type: ignore
+
+                job_calling_omconf()
+
+            run_job()
+
+    def test_success(self) -> None:
+        assert run_and_report(self.DemoFunctions.success_func) == 123
+
+    @mark.parametrize(
+        "demo_func, expected_traceback_regex",
+        [
+            param(
+                DemoFunctions.simple_error,
+                dedent(
+                    r"""
+                    Traceback \(most recent call last\):$
+                      File "[^"]+", line \d+, in run_and_report$
+                        return func\(\)$
+                      File "[^"]+", line \d+, in simple_error$
+                        assert False, "simple_err_msg"$
+                    AssertionError: simple_err_msg$
+                    assert False$
+                    """
+                ),
+                id="simple_failure_full_traceback",
+            ),
+            param(
+                DemoFunctions.run_job_wrapper,
+                dedent(
+                    r"""
+                    Traceback \(most recent call last\):$
+                      File "[^"]+", line \d+, in nested_error$
+                        assert False, "nested_err"$
+                    AssertionError: nested_err$
+                    assert False$
+                    Set the environment variable HYDRA_FULL_ERROR=1 for a complete stack trace\.$
+                    """
+                ),
+                id="strip_run_job_from_top_of_stack",
+            ),
+            param(
+                DemoFunctions.omegaconf_job_wrapper,
+                dedent(
+                    r"""
+                    Traceback \(most recent call last\):$
+                      File "[^"]+", line \d+, in job_calling_omconf$
+                        OmegaConf.resolve\(123\)  # type: ignore$
+                    ValueError: Invalid config type \(int\), expected an OmegaConf Container$
+                    Set the environment variable HYDRA_FULL_ERROR=1 for a complete stack trace\.$
+                    """
+                ),
+                id="strip_omegaconf_from_bottom_of_stack",
+            ),
+        ],
+    )
+    def test_failure(self, demo_func: Any, expected_traceback_regex: str) -> None:
+        mock_stderr = io.StringIO()
+        with raises(SystemExit, match="1"), patch("sys.stderr", new=mock_stderr):
+            run_and_report(demo_func)
+        mock_stderr.seek(0)
+        stderr_output = mock_stderr.read()
+        assert_regex_match(expected_traceback_regex, stderr_output)
+
+    def test_simplified_traceback_failure(self) -> None:
+        """
+        Test that a full traceback is printed in the case where an exception
+        occurs during the simplified traceback logic.
+        """
+        demo_func = self.DemoFunctions.omegaconf_job_wrapper
+        expected_traceback_regex = dedent(
+            r"""
+            Traceback \(most recent call last\):$
+              File "[^"]+", line \d+, in run_and_report$
+                return func\(\)$
+              File "[^"]+", line \d+, in omegaconf_job_wrapper$
+                run_job\(\)$
+              File "[^"]+", line \d+, in run_job$
+                job_calling_omconf\(\)$
+              File "[^"]+", line \d+, in job_calling_omconf$
+                OmegaConf.resolve(123)  # type: ignore
+              File "[^"]+", line \d+, in resolve$
+                raise ValueError(
+            ValueError: Invalid config type \(int\), expected an OmegaConf Container$
+            """
+        )
+        mock_stderr = io.StringIO()
+        with raises(SystemExit, match="1"), patch("sys.stderr", new=mock_stderr):
+            # patch `inspect.getmodule` so that an exception will occur in the
+            # simplified traceback logic:
+            with patch("inspect.getmodule", new=lambda *args: None):
+                run_and_report(demo_func)
+        mock_stderr.seek(0)
+        stderr_output = mock_stderr.read()
+        assert_regex_match(expected_traceback_regex, stderr_output)