pipe parent class around

dagster-io · Sep 22, 2023 · 2c59e3b · 2c59e3b
1 parent c3adde0
commit 2c59e3b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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 {},
     )