diff --git a/misc/makeboot.centos.py b/misc/makeboot.centos.py index 77baa060..a0d95820 100644 --- a/misc/makeboot.centos.py +++ b/misc/makeboot.centos.py @@ -39,7 +39,7 @@ def makeboot_tree(distribution, profiledir): os.link('/var/lib/confluent/public/site/site-initramfs.gz', trginitramfs) profileinfo = os.path.join(profiledir, 'profile.yaml') with open(profileinfo) as info: - profile = yaml.load(info) + profile = yaml.safe_load(info) cfgfile = os.path.join(efidir, 'grub.cfg') with open(cfgfile, 'w') as grubcfg: grubcfg.write('set timeout=5\n') @@ -57,4 +57,4 @@ def makeboot_tree(distribution, profiledir): if __name__ == '__main__': - makeboot_tree(sys.argv[1], sys.argv[2]) \ No newline at end of file + makeboot_tree(sys.argv[1], sys.argv[2])