From c2b052e5b7684b3cb1d7a33ba741118db06a6a6b Mon Sep 17 00:00:00 2001
From: marvinfriede <51965259+marvinfriede@users.noreply.github.com>
Date: Mon, 30 Sep 2024 16:46:18 -0500
Subject: [PATCH] Export `kill_timer` function

---
 src/dxtb/__init__.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/dxtb/__init__.py b/src/dxtb/__init__.py
index 87deeceb..b27ace98 100644
--- a/src/dxtb/__init__.py
+++ b/src/dxtb/__init__.py
@@ -22,7 +22,7 @@
 """
 
 # import timer first to get correct total time
-from dxtb._src.timing import timer
+from dxtb._src.timing import timer, kill_timer
 
 timer.start("Import")
 timer.start("PyTorch", parent_uid="Import")
@@ -54,7 +54,7 @@
 
 ###############################################################################
 
-# stop timers and remove from global namespace
+# stop timers and remove PyTorch from global namespace for cleaner API
 del torch
 timer.stop("dxtb")
 timer.stop("Import")
@@ -62,12 +62,17 @@
 ###############################################################################
 
 __all__ = [
-    "calculators",
     "components",
+    #
+    "calculators",
     "Calculator",
     "GFN1_XTB",
     "GFN2_XTB",
+    #
     "IndexHelper",
+    #
+    "kill_timer",
     "timer",
+    #
     "__version__",
 ]