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

from features.party_member import PartyMember


@offline
def get_party_members() -> DataFrame[PartyMember]:
    return scan_parquet(
        files="s3://q-chalk-test-data/surveyors-guild/party-member-rows/",
        columns=["id", "party_id", "adventurer_id"],
    )