From 2c59e3bc3ba4ac848bb7c0fe0a790b13340a94a5 Mon Sep 17 00:00:00 2001 From: JamieDeMaria Date: Thu, 21 Sep 2023 18:22:52 -0400 Subject: [PATCH] pipe parent class around --- python_modules/dagster/dagster/_core/ext/context.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python_modules/dagster/dagster/_core/ext/context.py b/python_modules/dagster/dagster/_core/ext/context.py index 1448694d6471c..2e099866a4e80 100644 --- a/python_modules/dagster/dagster/_core/ext/context.py +++ b/python_modules/dagster/dagster/_core/ext/context.py @@ -22,7 +22,7 @@ from dagster._core.definitions.partition_key_range import PartitionKeyRange from dagster._core.definitions.time_window_partitions import TimeWindow from dagster._core.execution.context.compute import OpExecutionContext -from dagster._core.execution.context.invocation import BoundOpExecutionContext +from dagster._core.execution.context.invocation import BoundExecutionContext class ExtMessageHandler: @@ -155,8 +155,8 @@ def build_external_execution_context_data( _convert_time_window(partition_time_window) if partition_time_window else None ), run_id=context.run_id, - job_name=None if isinstance(context, BoundOpExecutionContext) else context.job_name, - retry_number=0 if isinstance(context, BoundOpExecutionContext) else context.retry_number, + job_name=None if isinstance(context, BoundExecutionContext) else context.job_name, + retry_number=0 if isinstance(context, BoundExecutionContext) else context.retry_number, extras=extras or {}, )