Add utilities function

eth-cscs · Feb 29, 2024 · ec78946 · ec78946
1 parent 549a182
commit ec78946
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/firecrest/utilities.py b/firecrest/utilities.py
@@ -0,0 +1,11 @@
+import time
+from contextlib import contextmanager
+
+@contextmanager
+def time_block(label, logger):
+    start_time = time.time()
+    try:
+        yield
+    finally:
+        end_time = time.time()
+        logger.debug(f"{label} took {end_time - start_time:.6f} seconds")