12345678910111213141516171819202122 |
- """Provide functions commonly used in this dataset."""
- import yaml
- from pathlib import Path
- from ptpython.ipython import embed
- def get_marker_names(dataset):
- """Get all defined marker names from dataset specific yaml files."""
- yaml_path = Path(__file__).parent / ".." / "yamls" / f"{dataset}.yaml"
- with open(yaml_path, "r") as stream:
- try:
- dict_yaml = yaml.safe_load(stream)
- return ["BOLD_" + x["name"] for x in dict_yaml["markers"]]
- except yaml.YAMLError as exc:
- print(exc)
- if __name__ == "__main__":
- markers_piop1 = get_marker_names("PIOP1")
- embed()
|