[Bug] Hydra's exception formatting cannot handle exception raised in C extension

[YouJiacheng]

🐛 Bug

Description

I use hydra and av(PyAV), when av raise an error inside my hydra app, hydra cannot handle it correctly.
Because inspect.getmodule won't work for C extension.

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

Checklist

• I checked on the latest version of Hydra
• I created a minimal repro (See this for tips).

To reproduce

Minimal Code/Config snippet to reproduce

# foo.pyx
def foo():
    raise Exception

# main.py
import pyximport
pyximport.install() # to directly import pyx without `cythonize -i foo.pyx`

import hydra
import foo
@hydra.main(version_base='1.2')
def app(_):
    foo.foo()
app()

Stack trace/error message

Error executing job with overrides: []
An error occurred during Hydra's exception formatting:
AssertionError()
Traceback (most recent call last):
  File "/home/youjiacheng/miniconda3/envs/env/lib/python3.10/site-packages/hydra/_internal/utils.py", line 254, in run_and_report
    assert mdl is not None
AssertionError

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/youjiacheng/test/main.py", line 6, in <module>
    app()
  File "/home/youjiacheng/miniconda3/envs/env/lib/python3.10/site-packages/hydra/main.py", line 90, in decorated_main
    _run_hydra(
  File "/home/youjiacheng/miniconda3/envs/env/lib/python3.10/site-packages/hydra/_internal/utils.py", line 389, in _run_hydra
    _run_app(
  File "/home/youjiacheng/miniconda3/envs/env/lib/python3.10/site-packages/hydra/_internal/utils.py", line 452, in _run_app
    run_and_report(
  File "/home/youjiacheng/miniconda3/envs/env/lib/python3.10/site-packages/hydra/_internal/utils.py", line 296, in run_and_report
    raise ex
  File "/home/youjiacheng/miniconda3/envs/env/lib/python3.10/site-packages/hydra/_internal/utils.py", line 213, in run_and_report
    return func()
  File "/home/youjiacheng/miniconda3/envs/env/lib/python3.10/site-packages/hydra/_internal/utils.py", line 453, in <lambda>
    lambda: hydra.run(
  File "/home/youjiacheng/miniconda3/envs/env/lib/python3.10/site-packages/hydra/_internal/hydra.py", line 132, in run
    _ = ret.return_value
  File "/home/youjiacheng/miniconda3/envs/env/lib/python3.10/site-packages/hydra/core/utils.py", line 260, in return_value
    raise self._return_value
  File "/home/youjiacheng/miniconda3/envs/env/lib/python3.10/site-packages/hydra/core/utils.py", line 186, in run_job
    ret.return_value = task_function(task_cfg)
  File "/home/youjiacheng/test/main.py", line 5, in app
    foo.foo()
  File "foo.pyx", line 2, in foo.foo
    raise Exception
Exception

Expected Behavior

Following traceback is produced by renaming foo.pyx to foo.py.

Error executing job with overrides: []
Traceback (most recent call last):
  File "/home/youjiacheng/test/main.py", line 5, in app
    foo.foo()
  File "/home/youjiacheng/test/foo.py", line 2, in foo
    raise Exception
Exception

Set the environment variable HYDRA_FULL_ERROR=1 for a complete stack trace.

end: Optional[TracebackType] = tb
num_frames = 0
while end is not None:
    frame = end.tb_frame
    mdl = inspect.getmodule(frame)
    if mdl is not None:
        name = mdl.__name__
        if name.startswith("omegaconf."):
            break
    end = end.tb_next
    num_frames = num_frames + 1

System information

• Hydra Version : 1.2.0
• Python version : 3.10.4
• Virtual environment type and version : conda 4.13.0
• Operating system : CentOS Linux release 7.9.2009 (Core)