From a7884cef03d61ed28f33e1f0fe3828eef314de9b Mon Sep 17 00:00:00 2001
From: "Maarten A. Breddels" <maartenbreddels@gmail.com>
Date: Mon, 26 Feb 2024 15:06:28 +0100
Subject: [PATCH] fix: avoid stale data with use_task

---
 solara/tasks.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/solara/tasks.py b/solara/tasks.py
index 6a8c8616a..45949d8da 100644
--- a/solara/tasks.py
+++ b/solara/tasks.py
@@ -822,6 +822,8 @@ async def square():
 
     def wrapper(f):
         task_instance = solara.use_memo(lambda: task(f, prefer_threaded=prefer_threaded), dependencies=[])
+        # we always update the function so we do not have stale data in the function
+        task_instance.function = f  # type: ignore
 
         def _prestart():
             if dependencies is not None: