Fix issue with none results

PrefectHQ · Sep 3, 2024 · 6f81297 · 6f81297
1 parent e458363
commit 6f81297
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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