from chalk import DataFrame, offline
from chalk.sql import scan_parquet

from features.party import Party


@offline
def get_parties() -> DataFrame[Party]:
    return scan_parquet(
        files="s3://q-chalk-test-data/surveyors-guild/party-rows/",
        columns=["id", "name", "region", "formed_date"],
    )