diff --git a/cloud_archives/pv/passiv/passiv_monthly.py b/cloud_archives/pv/passiv/passiv_monthly.py index f3b9206..17ed598 100644 --- a/cloud_archives/pv/passiv/passiv_monthly.py +++ b/cloud_archives/pv/passiv/passiv_monthly.py @@ -83,7 +83,7 @@ def get_monthly_passiv_data(start_date: datetime, upload_to_hf: bool = True, ove @dg.asset( key=["pv", "passiv", "monthly_30min"], - auto_materialize_policy=dg.AutoMaterializePolicy.eager(), + auto_materialize_policy=dg.AutomationCondition.eager(), partitions_def=dg.TimeWindowPartitionsDefinition( fmt="%Y-%m", start="2010-01", @@ -104,7 +104,7 @@ def pv_passiv_monthly_30min(context: dg.AssetExecutionContext): @dg.asset( key=["pv", "passiv", "monthly_5min"], - auto_materialize_policy=dg.AutoMaterializePolicy.eager(), + auto_materialize_policy=dg.AutomationCondition.eager(), partitions_def=dg.TimeWindowPartitionsDefinition( fmt="%Y-%m", start="2018-01", diff --git a/cloud_archives/pv/passiv/passiv_year.py b/cloud_archives/pv/passiv/passiv_year.py index dbc8c9d..1210018 100644 --- a/cloud_archives/pv/passiv/passiv_year.py +++ b/cloud_archives/pv/passiv/passiv_year.py @@ -67,7 +67,7 @@ def get_yearly_passiv_data(start_date: datetime, upload_to_hf: bool = True, over @dg.asset( key=["pv", "passiv", "yearly_5min"], - auto_materialize_policy=dg.AutoMaterializePolicy.eager(), + auto_materialize_policy=dg.AutomationCondition.eager(), partitions_def=dg.TimeWindowPartitionsDefinition( fmt="%Y", start="2018", @@ -86,7 +86,7 @@ def pv_passiv_yearly_5min(context: dg.AssetExecutionContext): @dg.asset( key=["pv", "passiv", "yearly_30min"], - auto_materialize_policy=dg.AutoMaterializePolicy.eager(), + auto_materialize_policy=dg.AutomationCondition.eager(), partitions_def=dg.TimeWindowPartitionsDefinition( fmt="%Y", start="2010", diff --git a/local_archives/nwp/_generic_definitions_factory.py b/local_archives/nwp/_generic_definitions_factory.py index 3567af1..0d87a43 100644 --- a/local_archives/nwp/_generic_definitions_factory.py +++ b/local_archives/nwp/_generic_definitions_factory.py @@ -60,7 +60,7 @@ def make_definitions( @dg.asset( name="raw_archive", key_prefix=opts.key_prefix(), - auto_materialize_policy=dg.AutoMaterializePolicy.eager(), + auto_materialize_policy=dg.AutomationCondition.eager(), partitions_def=opts.partitions, check_specs=[ dg.AssetCheckSpec( @@ -168,7 +168,7 @@ def _raw_archive( name="zarr_archive", key_prefix=opts.key_prefix(), partitions_def=opts.partitions, - auto_materialize_policy=dg.AutoMaterializePolicy.eager(), + auto_materialize_policy=dg.AutomationCondition.eager(), ins={"raw_paths": dg.AssetIn(key=_raw_archive.key)}, io_manager_key="nwp_xr_zarr_io", compute_kind="process", diff --git a/local_archives/nwp/gfs/gfs.py b/local_archives/nwp/gfs/gfs.py index ed5c4b9..f8b06a2 100644 --- a/local_archives/nwp/gfs/gfs.py +++ b/local_archives/nwp/gfs/gfs.py @@ -13,7 +13,7 @@ name="zarr_daily_archive", description="Daily archive of GFS global NWP data", key_prefix=["nwp", "gfs", "global"], - auto_materialize_policy=dg.AutoMaterializePolicy.eager(), + auto_materialize_policy=dg.AutomationCondition.eager(), partitions_def=dg.DailyPartitionsDefinition( start_date="2015-01-15", end_offset=-2,