style: pre-commit fixes

Author: pre-commit-ci[bot]

arrakis/frion.py

@@ -16,6 +16,7 @@
 import numpy as np
 import pymongo
 from astropy.time import Time, TimeDelta
+from FRion import correct, predict
 from prefect import flow, get_run_logger, task
 from tqdm.auto import tqdm
 
@@ -28,7 +29,6 @@
 )
 from arrakis.utils.fitsutils import getfreq
 from arrakis.utils.pipeline import generic_parser, logo_str, workdir_arg_parser
-from FRion import correct, predict
 
 logger.setLevel(logging.INFO)
 TQDM_OUT = TqdmToLogger(logger, level=logging.INFO)

arrakis/utils/fitsutils.py

@@ -12,10 +12,10 @@
 from astropy.io import fits
 from astropy.utils.exceptions import AstropyWarning
 from astropy.wcs import WCS
+from FRion.correct import find_freq_axis
 from spectral_cube.utils import SpectralCubeWarning
 
 from arrakis.logger import logger
-from FRion.correct import find_freq_axis
 
 warnings.filterwarnings(action="ignore", category=SpectralCubeWarning, append=True)
 warnings.simplefilter("ignore", category=AstropyWarning)