From 06c074e3672d2a12234d182b06c2b336c55f3537 Mon Sep 17 00:00:00 2001 From: briantu Date: Thu, 17 Oct 2024 16:34:33 -0400 Subject: [PATCH] Fix for AssetCheckSummaryRecord --- .../dagster/dagster/_core/storage/event_log/base.py | 6 ++++-- 1 file changed, 4 insertions(+), 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 index ec298a5a75603..d5f3e2be3b7cc 100644 --- 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]