From d74d26df21591c3c0bc5a0b062ed7494db9e0f95 Mon Sep 17 00:00:00 2001 From: JamieDeMaria Date: Tue, 16 Jan 2024 13:13:41 -0500 Subject: [PATCH] DI context --- .../dagster/dagster/_core/execution/context/invocation.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/python_modules/dagster/dagster/_core/execution/context/invocation.py b/python_modules/dagster/dagster/_core/execution/context/invocation.py index c282e7915d215..219388150657a 100644 --- a/python_modules/dagster/dagster/_core/execution/context/invocation.py +++ b/python_modules/dagster/dagster/_core/execution/context/invocation.py @@ -22,6 +22,7 @@ from dagster._core.definitions.events import ( AssetMaterialization, AssetObservation, + CoercibleToAssetKey, ExpectationResult, UserEvent, ) @@ -800,6 +801,13 @@ def for_type(self, dagster_type: DagsterType) -> TypeCheckContext: def observe_output(self, output_name: str, mapping_key: Optional[str] = None) -> None: self.op_execution_context.observe_output(output_name=output_name, mapping_key=mapping_key) + def latest_materialization_for_upstream_asset( + self, key: CoercibleToAssetKey + ) -> Optional[AssetMaterialization]: + raise DagsterInvalidPropertyError( + _property_msg("latest_materialization_for_upstream_asset", "method") + ) + def _validate_resource_requirements( resource_defs: Mapping[str, ResourceDefinition], op_def: OpDefinition