Attempt test fix

vkottler · Aug 13, 2023 · a8ffaaf · a8ffaaf
1 parent 09b2e81
commit a8ffaaf
Showing 1 changed file with 16 additions and 18 deletions.
diff --git a/ifgen/generation/__init__.py b/ifgen/generation/__init__.py
@@ -26,21 +26,19 @@ def generate(root: Path, output: Path, config: Config) -> None:
         output.joinpath(subdir).mkdir(parents=True, exist_ok=True)
 
     pool = Pool()  # pylint: disable=consider-using-with
-    try:
-        for gen_name, generator in GENERATORS.items():
-            pool.map(
-                generator,
-                (
-                    GenerateTask(
-                        name,
-                        root,
-                        output.joinpath(gen_name, f"{name}.h"),
-                        data,
-                        config.data,
-                    )
-                    for name, data in config.data.get(gen_name, {}).items()
-                ),
-            )
-    finally:
-        pool.close()
-        pool.join()
+    for gen_name, generator in GENERATORS.items():
+        pool.map(
+            generator,
+            (
+                GenerateTask(
+                    name,
+                    root,
+                    output.joinpath(gen_name, f"{name}.h"),
+                    data,
+                    config.data,
+                )
+                for name, data in config.data.get(gen_name, {}).items()
+            ),
+        )
+    pool.close()
+    pool.join()