Handle attribute mapping properly

dagster-io · Dec 18, 2024 · 29c4c40 · 29c4c40
1 parent 9beb0c3
commit 29c4c40
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 = {