diff --git a/nwp/jobs.py b/nwp/jobs.py index 56368f6..890d852 100644 --- a/nwp/jobs.py +++ b/nwp/jobs.py @@ -1,6 +1,7 @@ import datetime as dt from dagster import ( + daily_partitioned_config, AssetSelection, DailyPartitionsDefinition, ScheduleDefinition, @@ -59,7 +60,7 @@ def build_config_on_runtime(model, run, delay=0): schedules.append(ScheduleDefinition(job=asset_job, cron_schedule="0 8 * * *")) -@partitioned_config(partitions_def=DailyPartitionsDefinition(start_date=dt.datetime(2021, 1, 1))) +@daily_partitioned_config(start_date=dt.datetime(2021, 1, 1)) def ecmwf_daily_partitioned_config(start: dt.datetime, _end: dt.datetime): return {"ops": {"nwp_consumer_docker_op": {"config": { "date_from": start.strftime("%Y-%m-%d"),