fix straggler

dagster-io · Jan 12, 2024 · 5331a95 · 5331a95
1 parent 8830d06
commit 5331a95
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]: