From 0cd7791d55beac2384da007d45489f9ffdf5725f Mon Sep 17 00:00:00 2001 From: JamieDeMaria Date: Thu, 4 Jan 2024 13:01:13 -0800 Subject: [PATCH] fix straggler --- python_modules/dagster/dagster/_core/execution/context/input.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python_modules/dagster/dagster/_core/execution/context/input.py b/python_modules/dagster/dagster/_core/execution/context/input.py index b4babbbad65cd..6b287fd235d2c 100644 --- a/python_modules/dagster/dagster/_core/execution/context/input.py +++ b/python_modules/dagster/dagster/_core/execution/context/input.py @@ -405,7 +405,7 @@ def asset_partitions_time_window(self) -> TimeWindow: "Tried to access asset_partitions_time_window, but the asset is not partitioned.", ) - return self.step_context.asset_partitions_time_window_for_input(self.name) + return self.step_context.asset_partitions_time_window_for_upstream(self.asset_key) @public def get_identifier(self) -> Sequence[str]: