Fix partitioning for ECMWF (#12)

openclimatefix · Sep 14, 2023 · 0df3884 · 0df3884
1 parent 9b5cc8f
commit 0df3884
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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"),