Fix for AssetCheckSummaryRecord

dagster-io · Oct 17, 2024 · 06c074e · 06c074e
1 parent 205c937
commit 06c074e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python_modules/dagster/dagster/_core/storage/event_log/base.py b/python_modules/dagster/dagster/_core/storage/event_log/base.py
@@ -160,9 +160,11 @@ class AssetCheckSummaryRecord(
 ):
     @classmethod
     def _blocking_batch_load(
-        cls, keys: Iterable[AssetCheckKey], instance: DagsterInstance
+        cls, keys: Iterable[AssetCheckKey], context: LoadingContext
     ) -> Iterable[Optional["AssetCheckSummaryRecord"]]:
-        records_by_key = instance.event_log_storage.get_asset_check_summary_records(list(keys))
+        records_by_key = context.instance.event_log_storage.get_asset_check_summary_records(
+            list(keys)
+        )
         return [records_by_key[key] for key in keys]