Handle attribute mapping properly

dagster-io · Dec 18, 2024 · 1efb94c · 1efb94c
1 parent c63cd60
commit 1efb94c
Show file tree

Hide file tree

Showing 3 changed files with 217 additions and 164 deletions.
diff --git a/python_modules/dagster/dagster/_core/definitions/asset_spec.py b/python_modules/dagster/dagster/_core/definitions/asset_spec.py
@@ -313,6 +313,7 @@ def replace_attributes(
         tags: Optional[Mapping[str, str]] = ...,
         kinds: Optional[Set[str]] = ...,
         partitions_def: Optional[PartitionsDefinition] = ...,
+        freshness_policy: Optional[FreshnessPolicy] = ...,
     ) -> "AssetSpec":
         """Returns a new AssetSpec with the specified attributes replaced."""
         current_tags_without_kinds = {