Try to fix an AttributeError related to pycodestyle 2.11.0

hhatto · Aug 13, 2023 · efbe9d4 · efbe9d4
1 parent 5b9110b
commit efbe9d4
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/autopep8.py b/autopep8.py
@@ -757,8 +757,12 @@ def fix_e225(self, result):
                 return
             if not check_syntax(fixed.lstrip()):
                 return
-            errors = list(
-                pycodestyle.missing_whitespace_around_operator(fixed, ts))
+            try:
+                _missing_whitespace = pycodestyle.missing_whitespace_around_operator
+            except AttributeError:
+                # pycodestyle >= 2.11.0
+                _missing_whitespace = pycodestyle.missing_whitespace
+            errors = list(_missing_whitespace(fixed, ts))
             for e in reversed(errors):
                 if error_code != e[1].split()[0]:
                     continue