From d01d5d2713db77daf01a5158fc72cb7eea9c94f0 Mon Sep 17 00:00:00 2001 From: JamieDeMaria Date: Fri, 22 Sep 2023 10:26:31 -0400 Subject: [PATCH] add interface back --- .../dagster/dagster/_core/execution/context/compute.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/python_modules/dagster/dagster/_core/execution/context/compute.py b/python_modules/dagster/dagster/_core/execution/context/compute.py index c7333670fcca0..64031c0a24ac6 100644 --- a/python_modules/dagster/dagster/_core/execution/context/compute.py +++ b/python_modules/dagster/dagster/_core/execution/context/compute.py @@ -12,7 +12,8 @@ cast, ) -# from dagster_ext import IContext +from dagster_ext import IContext + import dagster._check as check from dagster._annotations import deprecated, experimental, public from dagster._core.definitions.asset_check_spec import AssetCheckSpec @@ -1344,7 +1345,7 @@ def _get_deprecation_kwargs(attr: str): return deprecation_kwargs -class AssetExecutionContext(OpExecutionContext): +class AssetExecutionContext(OpExecutionContext, IContext): def __init__(self, op_execution_context: OpExecutionContext) -> None: self._op_execution_context = check.inst_param( op_execution_context, "op_execution_context", OpExecutionContext