From 55ae812dbf11f6568c73d5743aef0745715fb9fd Mon Sep 17 00:00:00 2001
From: thomas chaton <thomas@grid.ai>
Date: Mon, 8 Aug 2022 15:48:50 +0200
Subject: [PATCH] Resolve increased time. (#14074)

---
 src/lightning_app/CHANGELOG.md             | 2 ++
 src/lightning_app/utilities/proxies.py     | 2 +-
 tests/tests_app/core/test_lightning_app.py | 1 +
 3 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/lightning_app/CHANGELOG.md b/src/lightning_app/CHANGELOG.md
index 07927a1b01..78a4e370e7 100644
--- a/src/lightning_app/CHANGELOG.md
+++ b/src/lightning_app/CHANGELOG.md
@@ -27,3 +27,5 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 ### Fixed
 
 - Resolved a bug where the work statuses will grow quickly and be duplicated ([#13970](https://github.com/Lightning-AI/lightning/pull/13970))
+
+- Resolved a bug about a race condition when sending the work state through the caller_queue ([#14074](https://github.com/Lightning-AI/lightning/pull/14074))
diff --git a/src/lightning_app/utilities/proxies.py b/src/lightning_app/utilities/proxies.py
index 2c93a6c89f..99ad6e2aad 100644
--- a/src/lightning_app/utilities/proxies.py
+++ b/src/lightning_app/utilities/proxies.py
@@ -74,7 +74,7 @@ def _send_data_to_caller_queue(work: "LightningWork", caller_queue: "BaseQueue",
 
     data.update({"state": work_state})
     logger.debug(f"Sending to {work.name}: {data}")
-    caller_queue.put(data)
+    caller_queue.put(deepcopy(data))
 
     # Reset the calls entry.
     work_state["calls"] = calls
diff --git a/tests/tests_app/core/test_lightning_app.py b/tests/tests_app/core/test_lightning_app.py
index a3a15085b9..e6c715f87e 100644
--- a/tests/tests_app/core/test_lightning_app.py
+++ b/tests/tests_app/core/test_lightning_app.py
@@ -896,6 +896,7 @@ class SizeWork(LightningWork):
 
     def run(self, signal: int):
         self.counter += 1
+        assert len(self._calls) == 2
 
 
 class SizeFlow(LightningFlow):