# HG changeset patch
# User Jason R. Coombs <jaraco@jaraco.com>
# Date 2023-04-20 19:56:58
# Node ID e7ef11b75fdcacf7e3e1fa5494a35bb4718586d7
# Parent  eb01d3a65ad8bcd6db62df882cace8889df170be

library: incorporate demandimport into runpy invocation

diff --git a/mercurial/__main__.py b/mercurial/__main__.py
--- a/mercurial/__main__.py
+++ b/mercurial/__main__.py
@@ -1,4 +1,12 @@
+def run():
+    from . import demandimport
+
+    with demandimport.tracing.log('hg script'):
+        demandimport.enable()
+        from . import dispatch
+
+        dispatch.run()
+
+
 if __name__ == '__main__':
-    from . import dispatch
-
-    dispatch.run()
+    run()