diff --git a/matminer/featurizers/site/fingerprint.py b/matminer/featurizers/site/fingerprint.py index d1b4ab661..9dd3eac57 100644 --- a/matminer/featurizers/site/fingerprint.py +++ b/matminer/featurizers/site/fingerprint.py @@ -873,7 +873,7 @@ def load_cn_motif_op_params(): with open( os.path.join(os.path.dirname(pymatgen.analysis.local_env.__file__), "cn_opt_params.yaml"), ) as f: - return yaml.safe_load(f) + return yaml.YAML(typ="safe", pure=True).load(f) def load_cn_target_motif_op(): @@ -884,4 +884,4 @@ def load_cn_target_motif_op(): (dict) """ with open(os.path.join(os.path.dirname(__file__), "cn_target_motif_op.yaml")) as f: - return yaml.safe_load(f) + return yaml.YAML(typ="safe", pure=True).load(f)