Skip to content

Commit

Permalink
Improve compatibility with yaml
Browse files Browse the repository at this point in the history
  • Loading branch information
abelcarreras committed Mar 30, 2022
1 parent d031cf7 commit 2cec1db
Show file tree
Hide file tree
Showing 8 changed files with 8 additions and 8 deletions.
2 changes: 1 addition & 1 deletion cosymlib/file_io/tools.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ def print_input_info(initial_geometries, output=sys.stdout):

def add_extra_keywords(args, filename):
with open(filename, 'r') as stream:
input_parameters = yaml.load(stream, Loader=yaml.FullLoader)
input_parameters = yaml.safe_load(stream)
try:
for key, value in input_parameters.items():
if key.lower() in args:
Expand Down
2 changes: 1 addition & 1 deletion cosymlib/shape/tools.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ def get_reference_structure(label, central_atom=0):
if ideal_structures is None:
file_path = os.path.dirname(os.path.abspath(__file__)) + '/ideal_structures_center.yaml'
with open(file_path, 'r') as stream:
ideal_structures = yaml.load(stream, Loader=yaml.FullLoader)
ideal_structures = yaml.safe_load(stream)

try:
if central_atom == 0:
Expand Down
2 changes: 1 addition & 1 deletion scripts/cchir
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ args = parser.parse_args()

if args.yaml_input:
with open(args.yaml_input, 'r') as stream:
input_parameters = yaml.load(stream, Loader=yaml.FullLoader)
input_parameters = yaml.safe_load(stream)

for key, value in input_parameters.items():
if key.lower() in args:
Expand Down
2 changes: 1 addition & 1 deletion scripts/cosym
Original file line number Diff line number Diff line change
Expand Up @@ -159,7 +159,7 @@ args = parser.parse_args(sys.argv[1:])

if args.yaml_input:
with open(args.yaml_input, 'r') as stream:
input_parameters = yaml.load(stream, Loader=yaml.FullLoader)
input_parameters = yaml.safe_load(stream)

for key, value in input_parameters.items():
if key.lower() in args:
Expand Down
2 changes: 1 addition & 1 deletion scripts/esym
Original file line number Diff line number Diff line change
Expand Up @@ -104,7 +104,7 @@ args = parser.parse_args()

if args.yaml_input:
with open(args.yaml_input, 'r') as stream:
input_parameters = yaml.load(stream, Loader=yaml.FullLoader)
input_parameters = yaml.safe_load(stream)

for key, value in input_parameters.items():
if key.lower() in args:
Expand Down
2 changes: 1 addition & 1 deletion scripts/gsym
Original file line number Diff line number Diff line change
Expand Up @@ -94,7 +94,7 @@ args = parser.parse_args()

if args.yaml_input:
with open(args.yaml_input, 'r') as stream:
input_parameters = yaml.load(stream, Loader=yaml.FullLoader)
input_parameters = yaml.safe_load(stream)

for key, value in input_parameters.items():
if key.lower() in args:
Expand Down
2 changes: 1 addition & 1 deletion scripts/shape
Original file line number Diff line number Diff line change
Expand Up @@ -101,7 +101,7 @@ args = parser.parse_args()

if args.yaml_input:
with open(args.yaml_input, 'r') as stream:
input_parameters = yaml.load(stream, Loader=yaml.FullLoader)
input_parameters = yaml.safe_load(stream)

for key, value in input_parameters.items():
if key.lower() in args:
Expand Down
2 changes: 1 addition & 1 deletion scripts/shape_map
Original file line number Diff line number Diff line change
Expand Up @@ -104,7 +104,7 @@ args = parser.parse_args()

if args.yaml_input:
with open(args.yaml_input, 'r') as stream:
input_parameters = yaml.load(stream, Loader=yaml.FullLoader)
input_parameters = yaml.safe_load(stream)

for key, value in input_parameters.items():
if key.lower() in args:
Expand Down

0 comments on commit 2cec1db

Please sign in to comment.