Skip to content

parquet

ParquetDataLoader(path)

Bases: Dataset

DataLoader for Parquet files.

Initialize the ParquetDataLoader.

Parameters:

Name Type Description Default
path str | Path

Path to the Parquet file.

required
Source code in src/flowcean/environments/parquet.py
11
12
13
14
15
16
17
18
def __init__(self, path: str | Path) -> None:
    """Initialize the ParquetDataLoader.

    Args:
        path: Path to the Parquet file.
    """
    data = pl.read_parquet(path)
    super().__init__(data)