aboutsummaryrefslogtreecommitdiff
path: root/lava_dispatcher/device.py
diff options
context:
space:
mode:
Diffstat (limited to 'lava_dispatcher/device.py')
-rw-r--r--lava_dispatcher/device.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/lava_dispatcher/device.py b/lava_dispatcher/device.py
index 5ae4fcb68..797a5ffd4 100644
--- a/lava_dispatcher/device.py
+++ b/lava_dispatcher/device.py
@@ -105,12 +105,12 @@ class NewDevice(PipelineDevice):
if isinstance(target, str):
with open(target) as f_in:
data = f_in.read()
- data = yaml.load(data)
+ data = yaml.safe_load(data)
elif isinstance(target, dict):
data = target
else:
data = target.read()
- data = yaml.load(data)
+ data = yaml.safe_load(data)
if data is None:
raise ConfigurationError("Missing device configuration")
self.update(data)