start run_props as None

dagster-io · Nov 17, 2023 · 815d4e3 · 815d4e3
1 parent 36d3f15
commit 815d4e3
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/python_modules/dagster/dagster/_core/execution/context/compute.py b/python_modules/dagster/dagster/_core/execution/context/compute.py
@@ -1421,12 +1421,7 @@ def __init__(self, op_execution_context: OpExecutionContext) -> None:
             op_execution_context, "op_execution_context", OpExecutionContext
         )
 
-        self._run_props = RunProperties(
-            run_id=self._op_execution_context.run_id,
-            run_config=self._op_execution_context.run_config,
-            dagster_run=self._op_execution_context.run,
-            retry_number=self._op_execution_context.retry_number,
-        )
+        self._run_props = None
 
     @staticmethod
     def get() -> "AssetExecutionContext":
@@ -1441,6 +1436,13 @@ def op_execution_context(self) -> OpExecutionContext:
 
     @property
     def run_properties(self) -> RunProperties:
+        if self._run_props is None:
+            self._run_props = RunProperties(
+                run_id=self._op_execution_context.run_id,
+                run_config=self._op_execution_context.run_config,
+                dagster_run=self._op_execution_context.run,
+                retry_number=self._op_execution_context.retry_number,
+            )
         return self._run_props
 
     ######## Deprecated methods