FIX: compatibility with pip versions > 10.*.*

marionleborgne · Aug 28, 2018 · 82c2fac · 82c2fac
1 parent 81d18ff
commit 82c2fac
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -1,6 +1,12 @@
 from setuptools import setup, find_packages
-from pip.download import PipSession
-from pip.req import parse_requirements
+try:
+    from pip.req import parse_requirements
+except ImportError:
+    from pip._internal.req import parse_requirements
+try:
+    from pip.download import PipSession
+except ImportError:
+    from pip._internal.download import PipSession
 import os
 
 # Get __version__ and set other constants.