From da755d841189be563972d8ab02e5b29c2f162f60 Mon Sep 17 00:00:00 2001
From: Moritz <mr-tz@users.noreply.github.com>
Date: Fri, 26 Mar 2021 11:39:41 +0100
Subject: [PATCH] Update setup.py

---
 setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index effa60b7..b2d2bd6f 100644
--- a/setup.py
+++ b/setup.py
@@ -18,7 +18,6 @@ requirements = [
     "tabulate",
     "colorama",
     "termcolor",
-    "ruamel.yaml",
     "wcwidth",
     "ida-settings==2.1.0",
     "viv-utils==0.6.0",
@@ -28,6 +27,7 @@ if sys.version_info >= (3, 0):
     # py3
     requirements.append("halo")
     requirements.append("networkx")
+    requirements.append("ruamel.yaml==0.17.0")
     requirements.append("vivisect==1.0.0")
     requirements.append("smda==1.5.13")
 else:
@@ -36,6 +36,7 @@ else:
     requirements.append("halo==0.0.30")  # halo==0.0.30 is the last version to support py2.7
     requirements.append("vivisect==0.2.1")
     requirements.append("networkx==2.2")  # v2.2 is last version supported by Python 2.7
+    requirements.append("ruamel.yaml==0.16.13")  # last version tested with Python 2.7
     requirements.append("backports.functools-lru-cache")
 
 # this sets __version__