From 1056299071d3fc6f959dd04401f2492f31293e9a Mon Sep 17 00:00:00 2001 From: Dave Wallace Date: Mon, 16 Dec 2019 17:13:18 +0000 Subject: [PATCH] python3: fix wrk traffic profile YAML load Signed-off-by: Dave Wallace Change-Id: I3b7493221f3a0378b7f6cb0b0a21e8f18a0cc937 --- resources/tools/wrk/wrk_traffic_profile_parser.py | 4 ++-- 1 file 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 " -- 2.16.6