add interface back

dagster-io · Sep 22, 2023 · d01d5d2 · d01d5d2
1 parent 89f7433
commit d01d5d2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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