diff options
author | 2025-07-22 15:43:02 -0400 | |
---|---|---|
committer | 2025-07-22 15:43:02 -0400 | |
commit | 95fc4cf322b2141f76142e46e96e2846f5034edf (patch) | |
tree | 4804ea0456b83e3b2601c2c534ed865d67e458e7 /resources/unittest_template.py | |
parent | Update SCHEMA.md to support align property (diff) |
Add dynamic module loading fix for dataclasses
Diffstat (limited to 'resources/unittest_template.py')
-rw-r--r-- | resources/unittest_template.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/resources/unittest_template.py b/resources/unittest_template.py index d1944ee..5cdb8ec 100644 --- a/resources/unittest_template.py +++ b/resources/unittest_template.py @@ -52,6 +52,12 @@ def _load_user_module(result_writer, exit_wrapper: _EXIT_WRAPPER_TYPE) -> Module exit_wrapper(5) _module = ModuleType("module") + # It's necessary to manually add the module to the sys modules + # Dataclasses do not add themselves which causes issues for us with this type of dynamic loading + # if from __future__ import annotations is also used + # See: https://github.com/mkdocs/mkdocs/issues/3141 and https://github.com/sqlalchemy/alembic/issues/1419 + sys.modules[_module.__name__] = _module + exec(code, _module.__dict__) return _module |