Load the app before setting LIGHTNING_DISPATCHED (#16071)

Lightning-AI · Dec 15, 2022 · 1bdc6b3 · 1bdc6b3
1 parent eae56ee
commit 1bdc6b3
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/lightning_app/runners/runtime.py b/src/lightning_app/runners/runtime.py
@@ -55,9 +55,6 @@ def dispatch(
     from lightning_app.runners.runtime_type import RuntimeType
     from lightning_app.utilities.component import _set_flow_context
 
-    # Used to indicate Lightning has been dispatched
-    os.environ["LIGHTNING_DISPATCHED"] = "1"
-
     _set_flow_context()
 
     runtime_type = RuntimeType(runtime_type)
@@ -80,6 +77,8 @@ def dispatch(
         secrets=secrets,
         run_app_comment_commands=run_app_comment_commands,
     )
+    # Used to indicate Lightning has been dispatched
+    os.environ["LIGHTNING_DISPATCHED"] = "1"
     # a cloud dispatcher will return the result while local
     # dispatchers will be running the app in the main process
     return runtime.dispatch(open_ui=open_ui, name=name, no_cache=no_cache, cluster_id=cluster_id)