aboutsummaryrefslogtreecommitdiffstats
path: root/resources/tools/wrk/wrk_traffic_profile_parser.py
diff options
context:
space:
mode:
authorDave Wallace <dwallacelf@gmail.com>2019-12-16 17:13:18 +0000
committerDave Wallace <dwallacelf@gmail.com>2019-12-16 17:13:18 +0000
commit1056299071d3fc6f959dd04401f2492f31293e9a (patch)
tree15ccd3f7433ff8434b591149d22252e0a9624058 /resources/tools/wrk/wrk_traffic_profile_parser.py
parente8b2b5a90010999393cd92028dd7d274cc8852c7 (diff)
python3: fix wrk traffic profile YAML load
Signed-off-by: Dave Wallace <dwallacelf@gmail.com> Change-Id: I3b7493221f3a0378b7f6cb0b0a21e8f18a0cc937
Diffstat (limited to 'resources/tools/wrk/wrk_traffic_profile_parser.py')
-rw-r--r--resources/tools/wrk/wrk_traffic_profile_parser.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/resources/tools/wrk/wrk_traffic_profile_parser.py b/resources/tools/wrk/wrk_traffic_profile_parser.py
index 1994b6195d..f59210350b 100644
--- a/resources/tools/wrk/wrk_traffic_profile_parser.py
+++ b/resources/tools/wrk/wrk_traffic_profile_parser.py
@@ -20,7 +20,7 @@ See LLD for the structure of a wrk traffic profile.
from os.path import isfile
from pprint import pformat
-from yaml import load, YAMLError
+from yaml import safe_load, YAMLError
from robot.api import logger
from resources.tools.wrk.wrk_errors import WrkError
@@ -62,7 +62,7 @@ class WrkTrafficProfile:
try:
with open(self.profile_name, "r") as profile_file:
- self.traffic_profile = load(profile_file)
+ self.traffic_profile = safe_load(profile_file)
except IOError as err:
raise WrkError(
msg=f"An error occurred while opening the file "