From 38c46ec4d5cc7c81adf07ff0f25e62190f7fd437 Mon Sep 17 00:00:00 2001
From: benedikt-bartscher
 <31854409+benedikt-bartscher@users.noreply.github.com>
Date: Wed, 8 May 2024 22:31:20 +0200
Subject: [PATCH] copy background task marker (#3255)

---
 reflex/state.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/reflex/state.py b/reflex/state.py
index c9f0af5be..4663fc4de 100644
--- a/reflex/state.py
+++ b/reflex/state.py
@@ -39,6 +39,7 @@ from redis.asyncio import Redis
 from reflex import constants
 from reflex.base import Base
 from reflex.event import (
+    BACKGROUND_TASK_MARKER,
     Event,
     EventHandler,
     EventSpec,
@@ -579,6 +580,8 @@ class BaseState(Base, ABC, extra=pydantic.Extra.allow):
             closure=fn.__closure__,
         )
         newfn.__annotations__ = fn.__annotations__
+        if mark := getattr(fn, BACKGROUND_TASK_MARKER, None):
+            setattr(newfn, BACKGROUND_TASK_MARKER, mark)
         return newfn
 
     @staticmethod