Merge pull request #5 from boris-42/improve_main

Root trace point should collect full duration of imports
boris-42 · Apr 2, 2015 · 09a8939 · 09a8939
2 parents 04b8651 + 154d771
commit 09a8939
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/profimp/main.py b/profimp/main.py
@@ -47,7 +47,8 @@ def print_help():
 def trace_module(import_line):
     root_pt = tracer.init_stack()
     with tracer.patch_import():
-        exec(import_line)
+        with root_pt:
+            exec(import_line)
     return root_pt