add context interface to asset context

dagster-io · Sep 21, 2023 · 9258a58 · 9258a58
1 parent f20e4b2
commit 9258a58
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python_modules/dagster/dagster/_core/execution/context/compute.py b/python_modules/dagster/dagster/_core/execution/context/compute.py
@@ -12,6 +12,8 @@
     cast,
 )
 
+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
@@ -1342,7 +1344,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