add time window to DI contexts

dagster-io · Jan 26, 2024 · f87462e · f87462e
1 parent 90ba9e6
commit f87462e
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
@@ -617,6 +617,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],