Fix PipSession import for pip version 20

scidash · Jan 28, 2020 · 1aee48d · 1aee48d · appukuttan-shailesh · Feb 11, 2020
1 parent 4ac6465
commit 1aee48d
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -6,9 +6,12 @@
 try:
     from pip.req import parse_requirements
     from pip.download import PipSession
-except:
+except ImportError:
     from pip._internal.req import parse_requirements
-    from pip._internal.download import PipSession
+    try:
+        from pip._internal.download import PipSession
+    except ImportError:
+        from pip._internal.network.session import PipSession
 
 from setuptools import setup, find_packages