Add error handling

eyurtsev · Jul 16, 2024 · cecf13e · cecf13e
1 parent 33c49d4
commit cecf13e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/kor/extraction/api.py b/kor/extraction/api.py
@@ -24,8 +24,10 @@
 def set_verbose_context(verbose: bool):
     old_verbose = get_verbose()
     set_verbose(verbose)
-    yield
-    set_verbose(old_verbose)
+    try:
+        yield
+    finally:
+        set_verbose(old_verbose)
 
 
 async def _extract_from_document_with_semaphore(