Add module to sys.modules (#35)

pydantic · Aug 13, 2024 · 3108a2c · 3108a2c
1 parent 8519acc
commit 3108a2c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pytest_examples/run_code.py b/pytest_examples/run_code.py
@@ -49,6 +49,7 @@ def run_code(
         module.__dict__.update(module_globals)
     try:
         with insert_print:
+            sys.modules[spec.name] = module
             spec.loader.exec_module(module)
     except KeyboardInterrupt:
         print('KeyboardInterrupt in example')