Call atexit.register correctly

coveo · Dec 7, 2023 · 754e98f · 754e98f
1 parent 1ffaeaa
commit 754e98f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/coveo_stew/ci/mypy_runner.py b/coveo_stew/ci/mypy_runner.py
@@ -30,9 +30,9 @@ def _mypy_config_path(self) -> Optional[Path]:
             return None
 
         if self.set_config is True:
-            config_ref = importlib_resources.files("coveo_stew") / "package_resources/mypy.ini"
             stack = ExitStack()
-            atexit.register(stack)
+            atexit.register(stack.close)
+            config_ref = importlib_resources.files("coveo_stew") / "package_resources/mypy.ini"
             config_path = stack.enter_context(importlib_resources.as_file(config_ref))
             return config_path