ruff it up

arrakis/cutout.py

@@ -21,7 +21,6 @@
 from astropy.utils.exceptions import AstropyWarning
 from astropy.wcs.utils import skycoord_to_pixel
 from dask.distributed import Client, LocalCluster
-from distributed import get_client
 from prefect import flow, task, unmapped
 from spectral_cube import SpectralCube
 from spectral_cube.utils import SpectralCubeWarning

arrakis/linmos.py

@@ -15,7 +15,6 @@
 from astropy.utils.exceptions import AstropyWarning
 from dask.distributed import Client, LocalCluster
 from prefect import flow, task, unmapped
-from prefect.utilities.annotations import quote
 from racs_tools import beamcon_3D
 from spectral_cube.utils import SpectralCubeWarning
 from spython.main import Client as sclient

arrakis/makecat.py

@@ -17,7 +17,7 @@
 from astropy.stats import sigma_clip
 from astropy.table import Column, Table
 from dask.diagnostics import ProgressBar
-from prefect import flow, task, unmapped
+from prefect import flow, task
 from rmtable import RMTable
 from scipy.stats import lognorm, norm
 from tqdm import tqdm

arrakis/process_spice.py

@@ -3,18 +3,14 @@
 import logging
 import os
 from pathlib import Path
-from typing import Any, Optional, Tuple
 
 import configargparse
 import pkg_resources
 import yaml
 from astropy.time import Time
-from dask.distributed import Client, LocalCluster
-from dask_mpi import initialize
-from distributed.deploy.cluster import Cluster
 from prefect import flow
 from prefect.task_runners import BaseTaskRunner
-from prefect_dask import DaskTaskRunner, get_dask_client
+from prefect_dask import DaskTaskRunner
 
 from arrakis import (
     cleanup,
@@ -28,8 +24,7 @@
 )
 from arrakis.logger import logger
 from arrakis.utils.database import test_db
-from arrakis.utils.meta import class_for_name
-from arrakis.utils.pipeline import logo_str, performance_report_prefect
+from arrakis.utils.pipeline import logo_str
 
 
 @flow(name="Combining+Synthesis on Arrakis")