move comment

dagster-io · Dec 7, 2023 · 9c8cdb6 · 9c8cdb6
1 parent 3ef64cd
commit 9c8cdb6
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/python_modules/dagster/dagster/_core/execution/context/compute.py b/python_modules/dagster/dagster/_core/execution/context/compute.py
@@ -1361,13 +1361,15 @@ def get() -> "OpExecutionContext":
         return ctx.op_execution_context
 
 
-def _copy_docs_from_op_execution_context(obj):
-    setattr(obj, "__doc__", getattr(OpExecutionContext, obj.__name__).__doc__)
-    return obj
 ###############################
 ######## AssetExecutionContext
 ###############################
 
+def _copy_docs_from_op_execution_context(obj):
+    setattr(obj, "__doc__", getattr(OpExecutionContext, obj.__name__).__doc__)
+    return obj
+
+
 ALTERNATE_METHODS = {
     "run_id": "run_properties.run_id",
     "run": "run_properties.dagster_run",
@@ -1484,7 +1486,6 @@ def run_config(self) -> Mapping[str, object]:
     def retry_number(self):
         return self.op_execution_context.retry_number
 
-
     ########## pass-through to op context
 
     @public