Merge pull request #191 from Nnonexistent/pydocstyle_import_fix

Fix import to match pydocstyle v.1.1.0
prospector-dev · May 2, 2017 · 394859d · 394859d
2 parents 4b9aabf + d42dc91
commit 394859d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/prospector/tools/pep257/__init__.py b/prospector/tools/pep257/__init__.py
@@ -13,7 +13,11 @@ def dummy_log(*args, **kwargs):  # noqa
         pass
     pydocstyle.log.debug = dummy_log
 
-from pydocstyle import PEP257Checker, AllError
+try:
+    from pydocstyle.checker import PEP257Checker, AllError  # pydocstyle >= 1.1.0
+except ImportError:
+    from pydocstyle import PEP257Checker, AllError  # pydocstyle <= 1.1.0
+
 from prospector.message import Location, Message, make_tool_error_message
 from prospector.tools.base import ToolBase