Make it pip10 compatible

wizeline · Apr 17, 2018 · 54101f5 · 54101f5
1 parent f978a22
commit 54101f5
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -1,5 +1,10 @@
 import re
-from pip.req import parse_requirements
+
+try:  # for pip >= 10
+    from pip._internal.req import parse_requirements
+except ImportError:  # for pip <= 9.0.3
+    from pip.req import parse_requirements
+
 from setuptools import setup, find_packages