from collections.abc import Iterator

import polars as pl

def column_names() -> list[str]: ...
def parse_header_json(path: str) -> str: ...
def read_tracks(
    path: str, columns: list[str] | None = None, n_rows: int | None = None
) -> pl.DataFrame: ...

class RssaBatchReader:
    def __init__(
        self, path: str, batch_size: int, columns: list[str] | None = None
    ) -> None: ...
    def __iter__(self) -> Iterator[pl.DataFrame]: ...
    def __next__(self) -> pl.DataFrame: ...
