fix: import of matvis in fftvis

Author: Steven Murray

src/hera_sim/visibilities/fftvis.py

@@ -5,7 +5,6 @@
 import logging
 
 import numpy as np
-from matvis.core.beams import prepare_beam_unpolarized
 from pyuvdata import BeamInterface
 from pyuvdata import utils as uvutils
 
@@ -17,8 +16,9 @@
 
 try:
     import fftvis
+    from matvis.core.beams import prepare_beam_unpolarized
     HAVE_FFTVIS = True
-except ImportError:
+except ImportError:  # pragma: no cover
     HAVE_FFTVIS = False
     fftvis = None
 

src/hera_sim/visibilities/matvis.py

@@ -13,7 +13,7 @@
     HAVE_MATVIS = True
     if HAVE_GPU:
         from matvis import gpu
-except ImportError:
+except ImportError:  # pragma: no cover
     HAVE_GPU = False
     HAVE_MATVIS = False
     __version__ = None