From e2f68018cd9eceda87c11cdd25d2d8b288ec1cd5 Mon Sep 17 00:00:00 2001 From: Mark Nauwelaerts Date: Sat, 18 Jun 2016 09:51:21 +0200 Subject: [PATCH] Allow using git-remote-hg as module --- git-remote-hg | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/git-remote-hg b/git-remote-hg index adbb82f..ceadec2 100755 --- a/git-remote-hg +++ b/git-remote-hg @@ -1595,5 +1595,6 @@ def bye(): if is_tmp: shutil.rmtree(dirname) -atexit.register(bye) -sys.exit(main(sys.argv)) +if __name__ == '__main__': + atexit.register(bye) + sys.exit(main(sys.argv))