From 6f812973b2725817a9341e3b92e9e8ed8e658d97 Mon Sep 17 00:00:00 2001
From: Jeremiah Lowin <153965+jlowin@users.noreply.github.com>
Date: Mon, 2 Sep 2024 20:44:35 -0400
Subject: [PATCH] Fix issue with `none` results

---
 src/controlflow/tasks/task.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/controlflow/tasks/task.py b/src/controlflow/tasks/task.py
index 9bcc78e1..5662babd 100644
--- a/src/controlflow/tasks/task.py
+++ b/src/controlflow/tasks/task.py
@@ -650,6 +650,10 @@ def validate_result(self, raw_result: Any) -> T:
                 else:
                     result = self.result_type(raw_result)
 
+        # the raw result is None
+        else:
+            result = raw_result
+
             # Convert DataFrame schema back into pd.DataFrame object
             # if result_type == PandasDataFrame:
             #     import pandas as pd