Merge "Change tc_trex files to execute standalone tests"

[yardstick.git] / yardstick / benchmark / scenarios / lib / get_numa_info.py

diff --git a/yardstick/benchmark/scenarios/lib/get_numa_info.py b/yardstick/benchmark/scenarios/lib/get_numa_info.py
index 4e4a44d..75a9e35 100644 (file)
--- a/yardstick/benchmark/scenarios/lib/get_numa_info.py
+++ b/yardstick/benchmark/scenarios/lib/get_numa_info.py
@@ -13,7 +13,6 @@ from __future__ import absolute_import
 import logging
 import os
 
-import yaml
 from xml.etree import ElementTree as ET
 
 from yardstick import ssh
@@ -22,6 +21,7 @@ from yardstick.common import constants as consts
 from yardstick.common.utils import change_obj_to_dict
 from yardstick.common.openstack_utils import get_nova_client
 from yardstick.common.task_template import TaskTemplate
+from yardstick.common.yaml_loader import yaml_load
 
 LOG = logging.getLogger(__name__)
 
@@ -47,7 +47,7 @@ class GetNumaInfo(base.Scenario):
                                  self.options.get('file'))
 
         with open(node_file) as f:
-            nodes = yaml.safe_load(TaskTemplate.render(f.read()))
+            nodes = yaml_load(TaskTemplate.render(f.read()))
         self.nodes = {a['host_name']: a for a in nodes['nodes']}
 
     def run(self, result):