yaml

YamlDataLoader(path)

Bases: Dataset

DataLoader for YAML files.

Initialize the YamlDataLoader.

Parameters:

Name	Type	Description	Default
path	str | Path	Path to the YAML file.	required

Source code in src/flowcean/environments/yaml.py

def __init__(self, path: str | Path) -> None:
    """Initialize the YamlDataLoader.

    Args:
        path: Path to the YAML file.
    """
    data = pl.DataFrame(YAML(typ="safe").load(path))
    super().__init__(data)