Use yaml.safe_load instead of yaml.load