from dagster import Definitions, define_asset_job
from dagster_polars import PolarsParquetIOManager

from .assets import (asset_multi_1, asset_multi_2, asset_single_1,
                     asset_single_2)

# Define a job that includes both assets
daily_job = define_asset_job("daily_job", selection=[asset_multi_1, asset_multi_2])

vinyl = Definitions(
    assets=[asset_single_1, asset_multi_1, asset_single_2, asset_multi_2],
    resources={"polars_parquet_io_manager": PolarsParquetIOManager()},
    jobs=[daily_job],
)