diff --git a/src/quacc/wflow_tools/prefect_utils.py b/src/quacc/wflow_tools/prefect_utils.py index 5ec1923700..155827fdd2 100644 --- a/src/quacc/wflow_tools/prefect_utils.py +++ b/src/quacc/wflow_tools/prefect_utils.py @@ -4,7 +4,7 @@ from typing import TYPE_CHECKING, Any from prefect.futures import PrefectFuture -from prefect.results import BaseResult, ResultRecord +from prefect.results import ResultRecord from prefect.utilities.annotations import quote from prefect.utilities.collections import StopVisiting, visit_collection from typing_extensions import TypeVar @@ -56,9 +56,7 @@ def _collect_futures(futures, expr, context): future.wait() if future.state.is_completed(): result = future.state.result() - if isinstance(result, BaseResult): - result = result.get() - elif isinstance(result, ResultRecord): + if isinstance(result, ResultRecord): result = result.result results.append(result) else: @@ -121,9 +119,7 @@ def _collect_futures(futures, expr, context): future.wait() if future.state.is_completed(): result = future.state.result() - if isinstance(result, BaseResult): - result = result.get() - elif isinstance(result, ResultRecord): + if isinstance(result, ResultRecord): result = result.result results.append(result) else: