Fix mypy

corona10 · Nov 18, 2023 · b15f43d · b15f43d
1 parent 5682cf4
commit b15f43d
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/Tools/clinic/clinic.py b/Tools/clinic/clinic.py
@@ -1111,7 +1111,8 @@ def output_templates(
             if include:
                 clinic.add_include(include.filename, include.reason,
                                    condition=include.condition)
-
+        if f.critical_section:
+            clinic.add_include('pycore_critical_section.h', 'Py_BEGIN_CRITICAL_SECTION()')
         has_option_groups = parameters and (parameters[0].group or parameters[-1].group)
         simple_return = (f.return_converter.type == 'PyObject *'
                          and not f.critical_section)
@@ -5306,8 +5307,6 @@ def at_critical_section(self, *args: str) -> None:
             fail("Up to 2 critical section variables are supported")
         self.target_critical_section.extend(args)
         self.critical_section = True
-        if self.critical_section:
-            clinic.add_include('pycore_critical_section.h', 'Py_BEGIN_CRITICAL_SECTION()')
 
     def at_staticmethod(self) -> None:
         if self.kind is not CALLABLE: