import polars as pl
from bagelquant_data import DataLake, DatasetSpec

lake = DataLake.open("data")
daily = DatasetSpec("daily", "by_daily", calendar="trade_cal")
lake.ingest(
    daily,
    pl.DataFrame({"trade_date": ["20250102"], "ts_code": ["000001.SZ"], "close": [11.25]}),
)

close = lake.query.query("daily", source="custom", fields=["time", "asset_id", "close"])
print(close.collect())

Use uv sync to install the project and uv run pytest to run its tests.