fix callsite

dagster-io · Sep 6, 2023 · bcb05ee · bcb05ee
1 parent 3b2dd28
commit bcb05ee
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/python_modules/dagster/dagster/_core/execution/context/compute.py b/python_modules/dagster/dagster/_core/execution/context/compute.py
@@ -576,10 +576,11 @@ def partition_keys_for_asset(
                 of the dependecy version of the asset, not the partition of the asset as currently being materialized.
                 Defaults to False.
         """
-        return list(
-            self._step_execution_context.asset_partitions_subset_for_asset(
+        return self.partitions_def_for_asset(asset).get_partition_keys_in_range(
+            self._step_execution_context.asset_partition_key_range_for_asset(
                 asset, is_dependency=is_dependency
-            ).get_partition_keys()
+            ),
+            dynamic_partitions_store=self.instance,
         )
 
     @public