From 00d8db92dd8c92b3c7a859ac5f13f68305c74561 Mon Sep 17 00:00:00 2001
From: Jeremiah Lowin <153965+jlowin@users.noreply.github.com>
Date: Mon, 9 Sep 2024 19:46:56 -0400
Subject: [PATCH] Hotfix for prefect 3 compat

---
 src/controlflow/utilities/prefect.py | 14 +-------------
 1 file changed, 1 insertion(+), 13 deletions(-)

diff --git a/src/controlflow/utilities/prefect.py b/src/controlflow/utilities/prefect.py
index 5b9da445..0304f277 100644
--- a/src/controlflow/utilities/prefect.py
+++ b/src/controlflow/utilities/prefect.py
@@ -19,13 +19,11 @@
     TaskRunContext,
 )
 from prefect.events.schemas.events import Event
-from prefect.results import ResultFactory
 from prefect.states import (
     Cancelled,
     Completed,
     Failed,
     Running,
-    return_value_to_state,
 )
 from prefect.utilities.asyncutils import run_coro_as_sync
 from prefect.utilities.engine import (
@@ -223,17 +221,7 @@ def set_state(self, state: State) -> State:
         return new_state
 
     def succeed(self, result: Any):
-        if result is not None:
-            terminal_state = run_coro_as_sync(
-                return_value_to_state(
-                    result,
-                    result_factory=run_coro_as_sync(
-                        ResultFactory.from_autonomous_task(self._task)
-                    ),
-                )
-            )
-        else:
-            terminal_state = Completed()
+        terminal_state = Completed()
         self.set_state(terminal_state)
 
     def fail(self, error: Optional[str] = None):