Skip to content

Commit

Permalink
added return callable type P T
Browse files Browse the repository at this point in the history
  • Loading branch information
tgalvin committed Oct 11, 2024
1 parent 28695ad commit 2c05cf7
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 4 deletions.
2 changes: 1 addition & 1 deletion flint/configuration.py
Original file line number Diff line number Diff line change
Expand Up @@ -331,7 +331,7 @@ def wrapper_options_from_strategy(update_options_keyword: str):
update_options_keyword (str): The keyword option to update from the wrapped function
"""

def _wrapper(fn: Callable[P, T]) -> Callable:
def _wrapper(fn: Callable[P, T]) -> Callable[P, T]:
"""Decorator intended to allow options to be pulled from the
strategy file when function is called. See ``get_options_from_strategy``
for options that this function enables.
Expand Down
7 changes: 4 additions & 3 deletions flint/prefect/flows/continuum_pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -296,7 +296,8 @@ def process_science_fields(
strategy=unmapped(strategy),
mode="wsclean",
round_info="initial",
)
) # type: ignore

# TODO: This should be waited!
beam_summaries = task_create_beam_summary.map(
ms=preprocess_science_mss, imageset=wsclean_cmds
Expand Down Expand Up @@ -400,7 +401,7 @@ def process_science_fields(
strategy=unmapped(strategy),
mode="masking",
round_info=current_round,
)
) # type: ignore

wsclean_cmds = task_wsclean_imager.map(
in_ms=cal_mss,
Expand All @@ -409,7 +410,7 @@ def process_science_fields(
strategy=unmapped(strategy),
mode="wsclean",
round_info=current_round,
)
) # type: ignore
archive_wait_for.extend(wsclean_cmds)

# Do source finding on the last round of self-cal'ed images
Expand Down

0 comments on commit 2c05cf7

Please sign in to comment.