diff options
Diffstat (limited to 'lava_scheduler_app/scheduler.py')
-rw-r--r-- | lava_scheduler_app/scheduler.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/lava_scheduler_app/scheduler.py b/lava_scheduler_app/scheduler.py index 1c299d0ca..3d4f41ce2 100644 --- a/lava_scheduler_app/scheduler.py +++ b/lava_scheduler_app/scheduler.py @@ -143,7 +143,7 @@ def schedule_health_checks_for_device_type(logger, dt): def schedule_health_check(device, definition): user = User.objects.get(username="lava-health") - job = _create_pipeline_job(yaml.load(definition), user, [], + job = _create_pipeline_job(yaml.safe_load(definition), user, [], device_type=device.device_type, orig=definition, health_check=True) job.go_state_scheduled(device) @@ -206,7 +206,7 @@ def schedule_jobs_for_device(logger, device): if not job_tags.issubset(device_tags): continue - job_dict = yaml.load(job.definition) + job_dict = yaml.safe_load(job.definition) if 'protocols' in job_dict and 'lava-vland' in job_dict['protocols']: if not match_vlan_interface(device, job_dict): continue @@ -249,12 +249,12 @@ def transition_multinode_jobs(logger): # build a list of all devices in this group if sub_job.dynamic_connection: continue - definition = yaml.load(sub_job.definition) + definition = yaml.safe_load(sub_job.definition) devices[str(sub_job.id)] = definition['protocols']['lava-multinode']['role'] for sub_job in sub_jobs: # apply the complete list to all jobs in this group - definition = yaml.load(sub_job.definition) + definition = yaml.safe_load(sub_job.definition) definition['protocols']['lava-multinode']['roles'] = devices sub_job.definition = yaml.dump(definition) # transition the job and device |