Skip to content

Commit

Permalink
Update type hints (#2474)
Browse files Browse the repository at this point in the history
  • Loading branch information
Andrew-S-Rosen authored Sep 19, 2024
1 parent c082991 commit 1bf4a2f
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions src/quacc/wflow_tools/decorators.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,10 +4,13 @@

from collections.abc import Callable
from functools import partial, wraps
from typing import Any
from typing import TYPE_CHECKING, Any

from quacc.settings import change_settings_wrap

if TYPE_CHECKING:
from quacc.settings import QuaccSettings

Job = Callable[..., Any]
Flow = Callable[..., Any]
Subflow = Callable[..., Any]
Expand Down Expand Up @@ -627,7 +630,9 @@ def wrapper(
return wrapper


def _get_prefect_wrapped_flow(_func, settings, **kwargs):
def _get_prefect_wrapped_flow(
_func: Callable, settings: QuaccSettings, **kwargs
) -> Callable:
from prefect import flow as prefect_flow
from prefect.utilities.asyncutils import is_async_fn

Expand Down

0 comments on commit 1bf4a2f

Please sign in to comment.