update

Lightning-AI · Jun 18, 2024 · e72778e · e72778e
1 parent 2113bb7
commit e72778e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/lightning/app/core/app.py b/src/lightning/app/core/app.py
@@ -56,7 +56,7 @@
 from lightning.app.utilities.enum import AppStage, CacheCallsKeys
 from lightning.app.utilities.exceptions import CacheMissException, ExitAppException, LightningFlowException
 from lightning.app.utilities.layout import _collect_layout
-from lightning.app.utilities.proxies import ComponentDelta
+from lightning.app.utilities.proxies import ComponentDelta, unwrap
 from lightning.app.utilities.scheduler import SchedulerThread
 from lightning.app.utilities.tree import breadth_first
 from lightning.app.utilities.warnings import LightningFlowWarning
@@ -535,6 +535,7 @@ def _run(self) -> bool:
             for work in self.works:
                 _backend = work._backend
                 work._backend = None
+                work.run = unwrap(work.run)
                 self.backend._register_queues(self, work)
                 self.backend.create_work(self, work)
                 work._backend = _backend