Skip to content

Commit

Permalink
Fix Prefect (#2605)
Browse files Browse the repository at this point in the history
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
  • Loading branch information
Andrew-S-Rosen and pre-commit-ci[bot] authored Jan 5, 2025
1 parent 8f37195 commit d399abe
Showing 1 changed file with 3 additions and 7 deletions.
10 changes: 3 additions & 7 deletions src/quacc/wflow_tools/prefect_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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:
Expand Down Expand Up @@ -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:
Expand Down

0 comments on commit d399abe

Please sign in to comment.