diff --git a/flint/coadd/linmos.py b/flint/coadd/linmos.py index d6083ed1..307466f7 100644 --- a/flint/coadd/linmos.py +++ b/flint/coadd/linmos.py @@ -380,6 +380,7 @@ def generate_linmos_parameter_set( f"linmos.weighttype = Combined\n" f"linmos.weightstate = Inherent\n" f"linmos.cutoff = {cutoff}\n" + f"linmos.finalcutoff = 0.01\n" ) # Construct the holography section of the linmos parset parset += _get_holography_linmos_options(holofile=holofile, pol_axis=pol_axis) diff --git a/flint/prefect/common/imaging.py b/flint/prefect/common/imaging.py index 65691526..3aec7902 100644 --- a/flint/prefect/common/imaging.py +++ b/flint/prefect/common/imaging.py @@ -106,7 +106,8 @@ def task_potato_peel( @task def task_flag_ms_aoflagger(ms: FlagMS, container: Path) -> FlagMS: - extracted_ms = ms.ms if isinstance(ms, ApplySolutions) else ms + # Pirate believes the type ignore below is due to the decorated function and type alias + extracted_ms = MS.cast(ms=ms) # type: ignore extracted_ms = flag_ms_aoflagger(ms=extracted_ms, container=container) diff --git a/flint/prefect/flows/continuum_pipeline.py b/flint/prefect/flows/continuum_pipeline.py index 673421ed..58fb8392 100644 --- a/flint/prefect/flows/continuum_pipeline.py +++ b/flint/prefect/flows/continuum_pipeline.py @@ -181,6 +181,9 @@ def process_science_fields( preprocess_science_mss = task_copy_and_preprocess_casda_askap_ms.map( casda_ms=science_mss, output_directory=output_split_science_path ) + preprocess_science_mss = task_flag_ms_aoflagger.map( # type: ignore + ms=preprocess_science_mss, container=field_options.flagger_container + ) else: # TODO: This will likely need to be expanded should any # other calibration strategies get added diff --git a/pyproject.toml b/pyproject.toml index 44a9c529..11899cf6 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -39,6 +39,7 @@ pandas = "*" ConfigArgParse = "^1.7" fitscube = "^0.4.3" astroquery = "^0.4.7" +griffe = "0.48.0" [tool.poetry.group.dev.dependencies] black = "^23.3.0"