def test_yaml_save_load_roundtrip(extension):
data = {"some": ["data"]}
with NamedTemporaryFile() as f:
path = Path(f.name).with_suffix(extension)
save_to_yaml(data, path)
f.flush()
data_deserialized = load_yaml(path)
assert data == data_deserialized
def from_yaml(cls, path: Pathlike) -> "FeatureExtractor":
return cls.from_dict(load_yaml(path))
