From e7c5ea6bb43d01a70ad0b51ef2b5ecb14153c34e Mon Sep 17 00:00:00 2001
From: Ruslan Kuprieiev <kupruser@gmail.com>
Date: Thu, 17 Oct 2019 15:13:12 +0300
Subject: [PATCH] setup: use more flexible networkx reqs

https://github.com/iterative/dvc/commit/50fc721811a015142b476e90f8d198a99e88892f
---
 setup.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 518652dfe..a799d9fb1 100644
--- a/setup.py
+++ b/setup.py
@@ -49,7 +49,6 @@ class build_py(_build_py):
 
 
 install_requires = [
-    "networkx>=2.1,<2.3",
     "ply>=3.9",  # See https://github.com/pyinstaller/pyinstaller/issues/1945
     "configparser>=3.5.0",
     "future>=0.16.0",
@@ -79,6 +78,11 @@ install_requires = [
     "win-unicode-console>=0.5; sys_platform == 'win32'",
 ]
 
+if sys.version_info[0] == 2:
+    install_requires.append("networkx>=2.1,<2.3")
+else:
+    install_requires.append("networkx>=2.1,<2.4")
+
 # Extra dependencies for remote integrations
 
 gs = ["google-cloud-storage==1.19.0"]