Skip to content

Commit eec5701

Browse files
committed
refactor: Auto-patch AST within loading helpers
Issue #171: #171
1 parent 99bc9e6 commit eec5701

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

src/griffe/tests.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -139,6 +139,7 @@ def temporary_visited_package(
139139
Yields:
140140
A module.
141141
"""
142+
patch_ast()
142143
with temporary_pypackage(package, modules, init=init) as tmp_package:
143144
loader = GriffeLoader(search_paths=[tmp_package.tmpdir])
144145
yield loader.load_module(tmp_package.name)
@@ -171,6 +172,7 @@ def temporary_visited_module(
171172
Yields:
172173
The visited module.
173174
"""
175+
patch_ast()
174176
with temporary_pyfile(code, module_name=module_name) as (_, path):
175177
yield visit(
176178
module_name,

0 commit comments

Comments
 (0)