update

Lightning-AI · Jun 18, 2024 · 2113bb7 · 2113bb7
1 parent 71a1327
commit 2113bb7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/lightning/app/core/app.py b/src/lightning/app/core/app.py
@@ -533,8 +533,11 @@ def _run(self) -> bool:
             self._start_with_flow_works()
         else:
             for work in self.works:
+                _backend = work._backend
+                work._backend = None
                 self.backend._register_queues(self, work)
                 self.backend.create_work(self, work)
+                work._backend = _backend
 
         if self.should_publish_changes_to_api and self.api_publish_state_queue is not None:
             self.api_publish_state_queue.put((self.state_vars, self.status))