MNT: Actually enabled yaml_file as an input parameter