aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rwxr-xr-xhost/tests/devtest/uhd_test_base.py7
1 files changed, 6 insertions, 1 deletions
diff --git a/host/tests/devtest/uhd_test_base.py b/host/tests/devtest/uhd_test_base.py
index c2f987af4..00b0fd73b 100755
--- a/host/tests/devtest/uhd_test_base.py
+++ b/host/tests/devtest/uhd_test_base.py
@@ -17,7 +17,12 @@ import re
import time
import logging
from subprocess import Popen, PIPE
-import yaml
+# For what we're doing here, ruamel.yaml and yaml are copatible, and we'll use
+# whatever we can find
+try:
+ from ruamel import yaml
+except:
+ import yaml
from six import iteritems
from usrp_probe import get_usrp_list