Be sure to work with both versions of pydocstyle

prospector-dev · Oct 1, 2016 · d42dc91 · d42dc91
1 parent 87fdd76
commit d42dc91
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.checker 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