More ruff

scripts/casda_prepare.py

@@ -9,7 +9,6 @@
 import tarfile
 import time
 from glob import glob
-from typing import Dict
 
 import astropy.units as u
 import matplotlib.pyplot as plt
@@ -25,7 +24,6 @@
 from dask import delayed
 from dask.distributed import Client, LocalCluster
 from dask_mpi import initialize
-from IPython import embed
 from prefect_dask import get_dask_client
 from radio_beam import Beam
 from spectral_cube.cube_utils import convert_bunit
@@ -240,7 +238,7 @@ def convert_spectra(
         spectrum_table.table.add_column(new_col)
 
     outf = os.path.join(
-        spec_dir, os.path.basename(spectrum).replace(".dat", f"_polspec.fits")
+        spec_dir, os.path.basename(spectrum).replace(".dat", "_polspec.fits")
     )
     logger.info(f"Saving to {outf}")
     spectrum_table.write_FITS(outf, overwrite=True)

scripts/check_cutout.py

@@ -5,13 +5,7 @@
 import astropy.units as u
 import matplotlib.pyplot as plt
 from astropy.table import Table
-from astropy.visualization import (
-    ImageNormalize,
-    LogStretch,
-    MinMaxInterval,
-    SqrtStretch,
-    ZScaleInterval,
-)
+from astropy.visualization import ImageNormalize, MinMaxInterval, SqrtStretch
 from matplotlib.patches import Ellipse
 from spectral_cube import SpectralCube
 

scripts/compare_leakage.py

@@ -119,7 +119,7 @@ def interpolate(field, comp, beams, cutdir, septab, holofile, verbose=True):
 
             data[bm].update({f"{s}_holo": holo_spec})
             data[bm].update({f"{s}_image": im_spec})
-            data[bm].update({f"freq": freq})
+            data[bm].update({"freq": freq})
 
     try:
         outname = make_plot(data, comp, imfile)

scripts/compute_leakage.py

@@ -1,20 +1,15 @@
 #!/usr/bin/env python3
-import json
 
 import astropy
 import astropy.units as units
 import matplotlib.pyplot as plt
 import numpy as np
-import pymongo
 from astropy.coordinates import SkyCoord
-from astropy.io import fits
 from astropy.wcs import WCS
 from tqdm.auto import tqdm, trange
 
-from arrakis.logger import logger, logging
+from arrakis.logger import logger
 from arrakis.utils.database import get_db
-from arrakis.utils.fitsutils import getdata
-from arrakis.utils.json import MyEncoder
 
 
 def makesurf(start, stop, field, datadir, save_plots=True, data=None):
@@ -265,7 +260,7 @@ def cli():
     import getpass
 
     # Help string to be shown using the -h option
-    descStr = f"""
+    descStr = """
     Make leakage plots for a field.
 
     """

scripts/copy_cutouts.py

@@ -1,14 +1,8 @@
 import argparse
 import os
-import shlex
-import subprocess as sb
-from glob import glob
-from pprint import pprint
 
 import copy_data
-import numpy as np
 import spica
-from astropy.table import Table
 
 from arrakis.logger import logger, logging
 from arrakis.utils.io import try_mkdir
@@ -49,7 +43,7 @@ def main(field, dry_run=False, ncores=10):
 
 
 def cli():
-    descStr = f"""
+    descStr = """
     Copy data from RACS area to SPICE area'
     """
     parser = argparse.ArgumentParser(

scripts/copy_cutouts_askap.py

@@ -1,14 +1,8 @@
 import argparse
 import os
-import shlex
-import subprocess as sb
-from glob import glob
-from pprint import pprint
 
 import copy_data
-import numpy as np
 import spica
-from astropy.table import Table
 
 from arrakis.logger import logger, logging
 from arrakis.utils.io import try_mkdir
@@ -58,7 +52,7 @@ def main(field, dry_run=False, ncores=10):
 
 
 def cli():
-    descStr = f"""
+    descStr = """
     Copy data from RACS area to SPICE area'
     """
     parser = argparse.ArgumentParser(

scripts/copy_data.py

@@ -82,7 +82,7 @@ def main(
 
 
 def cli():
-    descStr = f"""
+    descStr = """
     Copy data from RACS area to SPICE area'
     """
     parser = argparse.ArgumentParser(

scripts/find_row.py

@@ -1,6 +1,5 @@
 #!/usr/bin/env python3
 import argparse
-import os
 from pathlib import Path
 
 from astropy.table import Table

scripts/find_sbid.py

@@ -1,9 +1,8 @@
 #!/usr/bin/env python3
 import argparse
-import os
 from pathlib import Path
 
-from astropy.table import Row, Table
+from astropy.table import Table
 
 from arrakis.logger import logger, logging
 

scripts/fix_dr1_cat.py

@@ -7,12 +7,10 @@
 
 import astropy.units as u
 import numpy as np
-import pkg_resources
 from astropy.coordinates import SkyCoord
 from astropy.table import Column, Table
 from astropy.time import Time
 from astropy.units import cds
-from IPython import embed
 from rmtable import RMTable
 from spica import SPICA
 
@@ -198,7 +196,7 @@ def main(cat: str, survey_dir: Path, epoch: int = 0):
     _, ext = os.path.splitext(cat)
     outfile = cat.replace(ext, f".corrected{ext}")
 
-    outfit = cat.replace(ext, f".corrected.leakage.pkl")
+    outfit = cat.replace(ext, ".corrected.leakage.pkl")
     with open(outfit, "wb") as f:
         pickle.dump(fit, f)
         logger.info(f"Wrote leakage fit to {outfit}")

scripts/fix_src_cat.py

@@ -6,7 +6,6 @@
 import numpy as np
 from astropy.coordinates import SkyCoord
 from astropy.table import Table
-from IPython import embed
 from tqdm.auto import tqdm
 
 from arrakis.logger import logger

scripts/make_links.py

@@ -28,7 +28,7 @@ def main(indir, outdir):
 
 
 def cli():
-    descStr = f"""
+    descStr = """
     Create symlinks to ASKAP cubes in one directory to another.
     """
     # Parse the command line options

scripts/spica.py

@@ -3,11 +3,9 @@
 import subprocess as sb
 from glob import glob
 from pathlib import Path
-from pprint import pprint
 
 import copy_data
 import numpy as np
-import pkg_resources
 from astropy.table import Table
 
 from arrakis.logger import logger, logging

scripts/tar_cubelets.py

@@ -6,7 +6,6 @@
 
 import dask
 from dask import delayed
-from dask.distributed import Client
 from tqdm.auto import tqdm
 
 from arrakis.logger import logger