Mercurial > hg

diff contrib/memory.py @ 31958:de5c9d0e02ea

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
atexit: switch to home-grown implementation
author Bryan O'Sullivan <bryano@fb.com>
date Tue, 11 Apr 2017 14:54:12 -0700
parents ff896733c66a
children 2372284d9457
line wrap: on
line diff
--- a/contrib/memory.py	Tue Apr 11 14:54:12 2017 -0700
+++ b/contrib/memory.py	Tue Apr 11 14:54:12 2017 -0700
@@ -12,7 +12,6 @@
 '''
 
 from __future__ import absolute_import
-import atexit
 
 def memusage(ui):
     """Report memory usage of the current process."""
@@ -29,4 +28,4 @@
                             for k, v in result.iteritems()]) + "\n")
 
 def extsetup(ui):
-    atexit.register(memusage, ui)
+    ui.atexit(memusage, ui)