add partition_time_window to DI context

dagster-io · Jan 31, 2024 · 0fc1636 · 0fc1636
1 parent b2de1c4
commit 0fc1636
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python_modules/dagster/dagster/_core/execution/context/invocation.py b/python_modules/dagster/dagster/_core/execution/context/invocation.py
@@ -646,6 +646,10 @@ def asset_partitions_time_window_for_output(self, output_name: str = "result") -
             Union[MultiPartitionsDefinition, TimeWindowPartitionsDefinition], partitions_def
         ).time_window_for_partition_key(self.partition_key)
 
+    @property
+    def partition_time_window(self) -> TimeWindow:
+        return self.asset_partitions_time_window_for_output()
+
     def add_output_metadata(
         self,
         metadata: Mapping[str, Any],