From 4392657a7181456e0a57ac23a82f8a37b76e97e1 Mon Sep 17 00:00:00 2001 From: Titouan Christophe Date: Mon, 16 Apr 2018 19:41:28 +0200 Subject: [PATCH] [fix] Pip imports for pip 10 --- setup.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 9f5e2990..3cb731d2 100644 --- a/setup.py +++ b/setup.py @@ -16,14 +16,20 @@ import datetime from distutils import log from distutils.core import Command -from pip.download import PipSession -from pip.req import parse_requirements from setuptools.command.develop import develop from setuptools.command.install import install from setuptools.command.sdist import sdist from setuptools import setup, find_packages from subprocess import check_output +import pip +if tuple(map(int, pip.__version__.split('.'))) >= (10, 0, 0): + from pip._internal.download import PipSession + from pip._internal.req import parse_requirements +else: + from pip.download import PipSession + from pip.req import parse_requirements + ROOT = os.path.realpath(os.path.join(os.path.dirname(__file__))) # When executing the setup.py, we need to be able to import ourselves, this